From dedb499efd4fc63a90eb4c5175d15e033a1881d7 Mon Sep 17 00:00:00 2001 From: jonas Date: Fri, 12 Nov 2021 23:00:07 +0100 Subject: [PATCH] :bug: fixing data loading bug --- tsflex/utils/data.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tsflex/utils/data.py b/tsflex/utils/data.py index 5896afe3..31111882 100644 --- 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 ]