fixed bug in metrics evaluation

attila-balint-kul · Jun 28, 2023 · f11528c · f11528c
1 parent f874419
commit f11528c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/enfobench/evaluation/evaluate.py b/src/enfobench/evaluation/evaluate.py
@@ -70,7 +70,7 @@ def evaluate_metric_on_forecasts(forecasts: pd.DataFrame, metric: Callable) -> p
     """
     metrics = {
         cutoff: evaluate_metric_on_forecast(group_df, metric)
-        for cutoff, group_df in forecasts.groupby("cutoff")
+        for cutoff, group_df in forecasts.groupby("cutoff_date")
     }
     metrics_df = pd.DataFrame.from_dict(metrics, orient="index", columns=["value"])
     return metrics_df