mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-27 15:37:44 +00:00
fix: correct the configuration inheritance relationship (#671)
* fix: Correct the configuration inheritance relationship * lint
This commit is contained in:
@@ -41,7 +41,7 @@ from rdagent.oai.llm_utils import md5_hash
|
||||
from rdagent.utils.workflow import wait_retry
|
||||
|
||||
|
||||
class EnvConf(BaseModel):
|
||||
class EnvConf(ExtendedBaseSettings):
|
||||
default_entry: str
|
||||
extra_volumes: dict = {}
|
||||
running_timeout_period: int = 600 # 10 minutes
|
||||
@@ -365,7 +365,7 @@ class MLECondaConf(CondaConf):
|
||||
|
||||
|
||||
## Docker Environment -----
|
||||
class DockerConf(EnvConf, ExtendedBaseSettings):
|
||||
class DockerConf(EnvConf):
|
||||
build_from_dockerfile: bool = False
|
||||
dockerfile_folder_path: Optional[Path] = (
|
||||
None # the path to the dockerfile optional path provided when build_from_dockerfile is False
|
||||
|
||||
+11
-1
@@ -6,13 +6,23 @@ class ConfUtils(unittest.TestCase):
|
||||
def test_conf(self):
|
||||
import os
|
||||
|
||||
from rdagent.utils.env import QlibDockerConf
|
||||
from rdagent.utils.env import EnvConf, QlibDockerConf
|
||||
|
||||
os.environ["MEM_LIMIT"] = "200g"
|
||||
assert QlibDockerConf().mem_limit == "200g" # base class will affect subclasses
|
||||
os.environ["QLIB_DOCKER_MEM_LIMIT"] = "300g"
|
||||
assert QlibDockerConf().mem_limit == "300g" # more accurate subclass will override the base class
|
||||
|
||||
os.environ["default_entry"] = "which python"
|
||||
os.environ["ENABLE_CACHE"] = "False"
|
||||
|
||||
assert EnvConf().enable_cache is False
|
||||
assert QlibDockerConf().enable_cache is False
|
||||
|
||||
os.environ["ENABLE_CACHE"] = "True"
|
||||
assert EnvConf().enable_cache is True
|
||||
assert QlibDockerConf().enable_cache is True
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user