fixed summary score bug with quantiles

supsi-dacd-isaac · Sep 24, 2024 · 4e38958 · 4e38958
1 parent 4615f87
commit 4e38958
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyforecaster/metrics.py b/pyforecaster/metrics.py
@@ -140,7 +140,11 @@ def scorer(estimator, X, y):
 
 def summary_score(x, t, score=rmse, agg_index=None):
     if len(x.shape) < 3:
-        if isinstance(x, pd.DataFrame) and isinstance(t, pd.DataFrame):
+        # if x is a pd.DataFrame and its columns are multiindex, just make sure t is a df
+        if isinstance(x, pd.DataFrame) and x.columns.nlevels > 1:
+            if not isinstance(t, pd.DataFrame):
+                t = pd.DataFrame(t, index=x.index)
+        elif isinstance(x, pd.DataFrame) and isinstance(t, pd.DataFrame):
             x.columns = t.columns
         elif isinstance(x, pd.DataFrame):
             t = pd.DataFrame(t, index=x.index, columns=x.columns)