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

fix: ds trace (#929)

Browse Source 
* fix: remove_ansi_codes
* chore: rename function
This commit is contained in:
Tim
2025-06-04 18:30:29 +08:00
committed by GitHub
parent 51a5247285
commit 543fb2c1ef
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/utils/workflow.py
+4 -4
View File
@@ -189,7 +189,7 @@ class LoopBase:
elif isinstance(e, self.withdraw_loop_error):
logger.warning(f"Withdraw loop {li} due to {e}")
# Back to previous loop
self.step_backward(li - 1)
self.withdraw_loop(li)
continue
else:
raise
@@ -215,8 +215,8 @@ class LoopBase:
self.dump(self.session_folder / f"{li}" / f"{si}_{name}") # save a snapshot after the session
def step_backward(self, li: int) -> None:
prev_session_dir = self.session_folder / str(li)
def withdraw_loop(self, loop_idx: int) -> None:
prev_session_dir = self.session_folder / str(loop_idx - 1)
prev_path = min(
(p for p in prev_session_dir.glob("*_*") if p.is_file()),
key=lambda item: int(item.name.split("_", 1)[0]),
@@ -233,7 +233,7 @@ class LoopBase:
# Overwrite current instance state
self.__dict__ = loaded.__dict__
else:
logger.error(f"No previous dump found at {prev_session_dir}, cannot withdraw loop {li}")
logger.error(f"No previous dump found at {prev_session_dir}, cannot withdraw loop {loop_idx}")
raise
def dump(self, path: str | Path) -> None: