parse time column in ForecastClient

attila-balint-kul · Jun 21, 2023 · 9d8be3d · 9d8be3d
1 parent 84da5fc
commit 9d8be3d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/enfobench/__version__.py b/src/enfobench/__version__.py
@@ -1 +1 @@
-__version__ = "0.0.4"
+__version__ = "0.0.5"
diff --git a/src/enfobench/evaluation/client.py b/src/enfobench/evaluation/client.py
@@ -60,5 +60,6 @@ def predict(
         if not response.ok:
             response.raise_for_status()
 
-        df = pd.DataFrame.from_records(response.json()['forecast'])
+        df = pd.DataFrame.from_records(response.json()["forecast"])
+        df['ds'] = pd.to_datetime(df['ds'])
         return df