fix: raise error in demo when no Metric in a Loop (#313)

* fix webpage when Metric is None * fix CI
microsoft · Sep 24, 2024 · e46a78e · e46a78e
1 parent f8f59ff
commit e46a78e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/rdagent/log/ui/app.py b/rdagent/log/ui/app.py
@@ -145,8 +145,11 @@ def get_msgs_until(end_func: Callable[[Message], bool] = lambda _: True):
                             state.alpha158_metrics = sms
 
                         # common metrics
-                        if msg.content.result is None and isinstance(state.scenario, DMModelScenario):
-                            state.metric_series.append(pd.Series([None], index=["AUROC"], name=f"Round {state.lround}"))
+                        if msg.content.result is None:
+                            if isinstance(state.scenario, DMModelScenario):
+                                state.metric_series.append(
+                                    pd.Series([None], index=["AUROC"], name=f"Round {state.lround}")
+                                )
                         else:
                             sms = msg.content.result
                             if isinstance(state.scenario, DMModelScenario):
@@ -155,8 +158,7 @@ def get_msgs_until(end_func: Callable[[Message], bool] = lambda _: True):
                                 state.scenario, (QlibModelScenario, QlibFactorFromReportScenario, QlibFactorScenario)
                             ):
                                 sms = sms.loc[QLIB_SELECTED_METRICS]
-                            elif isinstance(state.scenario, KGScenario):
-                                sms = sms.loc[["MCC"]]
+
                             sms.name = f"Round {state.lround}"
                             state.metric_series.append(sms)
                     elif "hypothesis generation" in tags: