fix: use CoSTEERSettings for DSRunnerCoSTEERSettings (#1096)

* refactor: use CoSTEERSettings for DSRunnerCoSTEERSettings

* lint

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:

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()