gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-27 23:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use CoSTEERSettings for DSRunnerCoSTEERSettings (#1096)

Browse Source 
* refactor: use CoSTEERSettings for DSRunnerCoSTEERSettings

* lint
This commit is contained in:
you-n-g
2025-07-20 10:53:09 +08:00
committed by GitHub
parent 81d0ac7185
commit 54d72e1dca
2 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/utils/test_conf.py
+18 -3
View File
@@ -1,12 +1,15 @@
import os
import unittest
from rdagent.app.data_science.conf import DS_RD_SETTING
from rdagent.components.coder.data_science.conf import DSCoderCoSTEERSettings
from rdagent.scenarios.data_science.dev.runner import DSRunnerCoSTEERSettings
from rdagent.utils.env import EnvConf, QlibDockerConf
class ConfUtils(unittest.TestCase):
def test_conf(self):
import os
from rdagent.utils.env import EnvConf, QlibDockerConf
os.environ["MEM_LIMIT"] = "200g"
os.environ["RUNNING_TIMEOUT_PERIOD"] = "None"
@@ -25,6 +28,18 @@ class ConfUtils(unittest.TestCase):
assert EnvConf().enable_cache is True
assert QlibDockerConf().enable_cache is True
def test_ds_costeer_conf(self):
os.environ["DS_CODER_COSTEER_MAX_SECONDS"] = "1000"
coder_conf = DSCoderCoSTEERSettings()
runner_conf = DSRunnerCoSTEERSettings()
print(coder_conf.max_seconds)
print(runner_conf.max_seconds)
assert coder_conf.max_seconds == 1000
# NOTE: coder's config should not affect runner's config
assert runner_conf.max_seconds == DS_RD_SETTING.full_timeout
os.environ["DS_RUNNER_COSTEER_MAX_SECONDS"] = "2000"
assert DSRunnerCoSTEERSettings().max_seconds == 2000
if __name__ == "__main__":
unittest.main()