Skip to content

Commit ad37417

Browse files
authored
fix: stop evolve if global timer is timeout (#1039)
1 parent 8bc12ea commit ad37417

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

rdagent/components/coder/CoSTEER/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
from rdagent.core.exception import CoderError
1818
from rdagent.core.experiment import Experiment
1919
from rdagent.log import rdagent_logger as logger
20+
from rdagent.oai.backend.base import RD_Agent_TIMER_wrapper
2021

2122

2223
class CoSTEER(Developer[Experiment]):
@@ -107,6 +108,9 @@ def develop(self, exp: Experiment) -> Experiment:
107108
if (datetime.now() - start_datetime).seconds > self.max_seconds:
108109
logger.info(f"Reached max time limit {self.max_seconds} seconds, stop evolving")
109110
break
111+
if RD_Agent_TIMER_wrapper.timer.started and RD_Agent_TIMER_wrapper.timer.is_timeout():
112+
logger.info("Global timer is timeout, stop evolving")
113+
break
110114

111115
if self.with_feedback and self.filter_final_evo:
112116
evo_exp = self._exp_postprocess_by_feedback(evo_exp, self.evolve_agent.evolving_trace[-1].feedback)

0 commit comments

Comments
 (0)