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

feat: refactor CoSTEER classes to use DSCoSTEER and update max seconds handling (#1156)

Browse Source 
* feat: refactor CoSTEER classes to use DSCoSTEER and update max seconds handling

* remove useless line

* enable time_ratio_limit_to_enable_hyperparameter_tuning
This commit is contained in:
Xu Yang
2025-08-05 18:16:10 +08:00
committed by GitHub
parent 50916fa6c9
commit 0284b73596
13 changed files with 53 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/ensemble/__init__.py
+2 -3
View File
@@ -16,7 +16,6 @@ from pathlib import Path
from jinja2 import Environment, StrictUndefined
from rdagent.app.data_science.conf import DS_RD_SETTING
from rdagent.components.coder.CoSTEER import CoSTEER
from rdagent.components.coder.CoSTEER.evaluators import (
CoSTEERMultiEvaluator,
CoSTEERSingleFeedback,
@@ -30,6 +29,7 @@ from rdagent.components.coder.CoSTEER.knowledge_management import (
from rdagent.components.coder.data_science.conf import DSCoderCoSTEERSettings
from rdagent.components.coder.data_science.ensemble.eval import EnsembleCoSTEEREvaluator
from rdagent.components.coder.data_science.ensemble.exp import EnsembleTask
from rdagent.components.coder.data_science.share.ds_costeer import DSCoSTEER
from rdagent.core.exception import CoderError
from rdagent.core.experiment import FBWorkspace
from rdagent.core.scenario import Scenario
@@ -141,7 +141,7 @@ class EnsembleMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
return evo
class EnsembleCoSTEER(CoSTEER):
class EnsembleCoSTEER(DSCoSTEER):
def __init__(
self,
scen: Scenario,
@@ -160,6 +160,5 @@ class EnsembleCoSTEER(CoSTEER):
evolving_version=2,
scen=scen,
max_loop=DS_RD_SETTING.coder_max_loop,
max_seconds=scen.real_debug_timeout() * settings.max_seconds_multiplier,
**kwargs,
)