bugfixes in ForecastClient

attila-balint-kul · Jun 22, 2023 · 4547681 · 4547681
1 parent f958618
commit 4547681
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 6 deletions.
diff --git a/src/enfobench/__version__.py b/src/enfobench/__version__.py
@@ -1 +1 @@
-__version__ = "0.0.9"
+__version__ = "0.0.10"
diff --git a/src/enfobench/evaluation/_cross_validate.py b/src/enfobench/evaluation/_cross_validate.py
@@ -80,7 +80,7 @@ def cross_validate(
     forecasts = []
     for cutoff in tqdm(cutoff_dates):
         # make sure that there is no data leakage
-        history = y.loc[y.ds <= cutoff, ["ds", "y"]]
+        history = y.loc[y.index <= cutoff]
 
         forecast = model.predict(
             horizon_length,

diff --git a/src/enfobench/evaluation/client.py b/src/enfobench/evaluation/client.py
@@ -37,7 +37,7 @@ def environment(self) -> EnvironmentInfo:
     def predict(
         self,
         horizon: int,
-        y: pd.DataFrame,
+        y: pd.Series,
         # X: pd.DataFrame,
         level: list[int] | None = None,
     ) -> pd.DataFrame:
@@ -47,8 +47,9 @@ def predict(
         if level is not None:
             params["level"] = level
 
+        y_df = y.rename_axis('ds').reset_index()
         files = {
-            "y": to_buffer(y),
+            "y": to_buffer(y_df),
             # "X": to_buffer(X),
         }
 

diff --git a/src/enfobench/utils.py b/src/enfobench/utils.py
@@ -1,5 +1,3 @@
-from __future__ import annotations
-
 import pandas as pd
 from pandas import Timedelta