gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-06 19:47:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct the configuration inheritance relationship (#671)

Browse Source 
* fix: Correct the configuration inheritance relationship

* lint
This commit is contained in:
you-n-g
2025-03-12 18:49:00 +08:00
committed by GitHub
parent cf86d8feeb
commit 55ac3e4a49
2 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/utils/test_conf.py
+11 -1
View File
@@ -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()