fixed issue with serializing forecast response in client

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

Hide file tree

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