diff --git a/rdagent/scenarios/data_science/dev/runner/__init__.py b/rdagent/scenarios/data_science/dev/runner/__init__.py index a9cb8fba..5c1820a6 100644 --- a/rdagent/scenarios/data_science/dev/runner/__init__.py +++ b/rdagent/scenarios/data_science/dev/runner/__init__.py @@ -3,7 +3,7 @@ import pandas as pd from rdagent.app.data_science.conf import DS_RD_SETTING from rdagent.components.coder import CoSTEER from rdagent.components.coder.CoSTEER import CoSTEER -from rdagent.components.coder.CoSTEER.config import CoSTEER_SETTINGS +from rdagent.components.coder.CoSTEER.config import CoSTEERSettings from rdagent.components.coder.CoSTEER.evaluators import ( CoSTEERMultiEvaluator, CoSTEERSingleFeedback, @@ -14,7 +14,6 @@ from rdagent.components.coder.CoSTEER.evolving_strategy import ( MultiProcessEvolvingStrategy, ) from rdagent.components.coder.CoSTEER.task import CoSTEERTask -from rdagent.components.coder.data_science.conf import DSCoderCoSTEERSettings from rdagent.components.coder.data_science.share.eval import ModelDumpEvaluator from rdagent.core.exception import RunnerError from rdagent.core.scenario import Scenario @@ -26,7 +25,7 @@ from rdagent.utils.agent.tpl import T from rdagent.utils.workflow import wait_retry -class DSRunnerCoSTEERSettings(DSCoderCoSTEERSettings): +class DSRunnerCoSTEERSettings(CoSTEERSettings): """Data Science CoSTEER settings""" class Config: diff --git a/test/utils/test_conf.py b/test/utils/test_conf.py index 3a5a38a7..086f13e3 100644 --- a/test/utils/test_conf.py +++ b/test/utils/test_conf.py @@ -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()