From d6990a12b8cc2f3a467f2147703da88263da4e0c Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Thu, 24 Apr 2025 10:39:31 -0600 Subject: [PATCH] If not use the session stored timer, need to replace it with the default timer (#825) --- rdagent/utils/workflow.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/rdagent/utils/workflow.py b/rdagent/utils/workflow.py index da86b301..c0baa7d8 100644 --- a/rdagent/utils/workflow.py +++ b/rdagent/utils/workflow.py @@ -227,9 +227,14 @@ class LoopBase: max_loop = max(session.loop_trace.keys()) logger.storage.truncate(time=session.loop_trace[max_loop][-1].end) - if session.timer.started and replace_timer: - RD_Agent_TIMER_wrapper.replace_timer(session.timer) - RD_Agent_TIMER_wrapper.timer.restart_by_remain_time() + if session.timer.started: + if replace_timer: + RD_Agent_TIMER_wrapper.replace_timer(session.timer) + RD_Agent_TIMER_wrapper.timer.restart_by_remain_time() + else: + # Use the default timer to replace the session timer + session.timer = RD_Agent_TIMER_wrapper.timer + return session