From e46a78eb69271cb19978aab2f3b976c2870ca082 Mon Sep 17 00:00:00 2001 From: XianBW <36835909+XianBW@users.noreply.github.com> Date: Tue, 24 Sep 2024 16:14:01 +0800 Subject: [PATCH] fix: raise error in demo when no Metric in a Loop (#313) * fix webpage when Metric is None * fix CI --- rdagent/log/ui/app.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/rdagent/log/ui/app.py b/rdagent/log/ui/app.py index 2d453cca..1408fa50 100644 --- 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: