From a697ec0069939ab0f225f6c33f6c7fae6e647e9c Mon Sep 17 00:00:00 2001 From: Aleksey Veresov Date: Mon, 14 Oct 2024 14:45:48 +0200 Subject: [PATCH] Fix --- python/hsfs/engine/python.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/hsfs/engine/python.py b/python/hsfs/engine/python.py index 44cf505fc..a287ba4d1 100644 --- a/python/hsfs/engine/python.py +++ b/python/hsfs/engine/python.py @@ -1473,9 +1473,9 @@ def _start_offline_materialization(offline_write_options: Dict[str, Any]) -> boo def _convert_feature_log_to_df(feature_log, cols) -> pd.DataFrame: if feature_log is None and cols: return pd.DataFrame(columns=cols) - if not (isinstance(feature_log, (list, pd.DataFrame, pl.DataFrame))) or ( + if not (isinstance(feature_log, (list, pd.DataFrame, pl.DataFrame)) or ( HAS_NUMPY and isinstance(feature_log, np.ndarray) - ): + )): raise ValueError(f"Type '{type(feature_log)}' not accepted") if isinstance(feature_log, list) or ( HAS_NUMPY and isinstance(feature_log, np.ndarray)