🐛 fixing data loading bug

predict-idlab · Nov 12, 2021 · dedb499 · dedb499
1 parent fecaa4d
commit dedb499
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tsflex/utils/data.py b/tsflex/utils/data.py
@@ -174,6 +174,8 @@ def load_empatica_data(f_names: Union[str, List[str]]) -> List[pd.DataFrame]:
     url = "https://github.com/predict-idlab/tsflex/raw/main/examples/data/empatica/"
     f_names = [f_names] if isinstance(f_names, str) else f_names
     return [
-        pd.read_parquet(url + f"{f_name.lower()}.parquet").set_index("timestamp")
+        pd.read_parquet(
+            url + f"{f_name.lower()}.parquet", engine="fastparquet"
+        ).set_index("timestamp")
         for f_name in f_names
     ]