From 621391af16492407e1710e07c5307ad16dec0ba9 Mon Sep 17 00:00:00 2001 From: Martin Stancsics Date: Mon, 23 Sep 2024 10:44:14 +0200 Subject: [PATCH] Remove unnecessary argument --- src/tabmat/categorical_matrix.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/tabmat/categorical_matrix.py b/src/tabmat/categorical_matrix.py index 4de40a3a..f128b15b 100644 --- a/src/tabmat/categorical_matrix.py +++ b/src/tabmat/categorical_matrix.py @@ -233,7 +233,7 @@ def _extract_codes_and_categories_pandas(cat_vec) -> tuple[np.ndarray, np.ndarra categories = cat_vec.cat.categories indices = cat_vec.cat.codes.to_numpy() else: - indices, categories = pd.factorize(cat_vec, sort=True, use_na_sentinel=True) + indices, categories = pd.factorize(cat_vec, sort=True) return indices, categories.to_numpy() @@ -246,8 +246,8 @@ def _extract_codes_and_categories_polars(cat_vec) -> tuple[np.ndarray, np.ndarra def _extract_codes_and_categories_numpy(cat_vec) -> tuple[np.ndarray, np.ndarray]: - if pd: - indices, categories = pd.factorize(cat_vec, sort=True, use_na_sentinel=True) + if pd and pd.__version__ >= "1.5.0": + indices, categories = pd.factorize(cat_vec, sort=True) else: indices, categories = _factorize(cat_vec) return indices, categories