fix display bug in webapp (#440)

microsoft · Oct 18, 2024 · c3fa245 · c3fa245
1 parent 24aec11
commit c3fa245
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/rdagent/log/ui/app.py b/rdagent/log/ui/app.py
@@ -518,6 +518,11 @@ def research_window():
             # loaded model exp
             with c2:
                 if mem := state.msgs[round]["d.load_experiment"]:
+                    # 'load_experiment' should in 'r' now, but old version trace may in 'd', so we need to check both
+                    # TODO: modify the way to get one message with a specific tag like 'load_experiment' in the future
+                    me: QlibModelExperiment = mem[0].content
+                    tasks_window(me.sub_tasks)
+                elif mem := state.msgs[round]["r.load_experiment"]:
                     me: QlibModelExperiment = mem[0].content
                     tasks_window(me.sub_tasks)