From 58229b5abec818285853cd4e123bdfdeaa7439e5 Mon Sep 17 00:00:00 2001 From: felunka Date: Tue, 5 Sep 2023 13:18:25 +0200 Subject: [PATCH] Fix is_categorical_dtype deprecation --- category_encoders/utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/category_encoders/utils.py b/category_encoders/utils.py index 580a1eaf..32fa7172 100644 --- a/category_encoders/utils.py +++ b/category_encoders/utils.py @@ -6,6 +6,7 @@ import pandas as pd import numpy as np import sklearn.base +from pandas.core.dtypes.dtypes import CategoricalDtype from sklearn.base import BaseEstimator, TransformerMixin from sklearn.exceptions import NotFittedError from typing import Dict, List, Optional, Union @@ -25,7 +26,7 @@ def convert_cols_to_list(cols): return list(cols) elif isinstance(cols, tuple): return list(cols) - elif pd.api.types.is_categorical_dtype(cols): + elif isinstance(cols, CategoricalDtype): return cols.astype(object).tolist() return cols @@ -47,7 +48,7 @@ def get_obj_cols(df): def is_category(dtype): - return pd.api.types.is_categorical_dtype(dtype) + return isinstance(dtype, CategoricalDtype) def convert_inputs(X, y, columns=None, index=None, deep=False):