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

fix current_selection maybe empty (#1125)

Browse Source
This commit is contained in:
XianBW
2025-07-29 11:56:59 +08:00
committed by GitHub
parent 60cae0f588
commit 8790503e8d
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/log/ui/ds_trace.py
+10 -5
View File
@@ -522,11 +522,16 @@ def main_win(loop_id, llm_data=None):
if "running" in loop_data:
# get last SOTA_exp_to_submit
current_trace = loop_data["record"]["trace"]
parent_idxs = current_trace.get_parents(current_trace.get_current_selection()[0])
if len(parent_idxs) >= 2 and hasattr(current_trace, "idx2loop_id"):
parent_idx = parent_idxs[-2]
parent_loop_id = current_trace.idx2loop_id[parent_idx]
sota_exp = state.data[parent_loop_id]["record"].get("sota_exp_to_submit", None)
current_selection = current_trace.get_current_selection()
if len(current_selection) > 0: # TODO: Why current_selection can be "()"?
current_idx = current_selection[0]
parent_idxs = current_trace.get_parents(current_idx)
if len(parent_idxs) >= 2 and hasattr(current_trace, "idx2loop_id"):
parent_idx = parent_idxs[-2]
parent_loop_id = current_trace.idx2loop_id[parent_idx]
sota_exp = state.data[parent_loop_id]["record"].get("sota_exp_to_submit", None)
else:
sota_exp = None
else:
sota_exp = None