From bcc222b15edf66e243adba94d1175fa7c3ddd147 Mon Sep 17 00:00:00 2001 From: Vyacheslav Morov Date: Thu, 9 Dec 2021 16:22:39 +0300 Subject: [PATCH] Fix cat_feature_names filtration #2. --- evidently/analyzers/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/evidently/analyzers/utils.py b/evidently/analyzers/utils.py index e4bab5f8c3..cc0ac4136e 100644 --- a/evidently/analyzers/utils.py +++ b/evidently/analyzers/utils.py @@ -70,9 +70,9 @@ def process_columns(dataset: pd.DataFrame, column_mapping: ColumnMapping): cat_feature_names = column_mapping.categorical_features if cat_feature_names is None: - cat_feature_names = list(set(dataset.select_dtypes([np.object]).columns) - set(utility_columns)) + cat_feature_names = list(set(dataset.select_dtypes(exclude=[np.number]).columns) - set(utility_columns)) else: - cat_feature_names = dataset[cat_feature_names].select_dtypes([np.object]).columns.tolist() + cat_feature_names = dataset[cat_feature_names].columns.tolist() return DatasetColumns( DatasetUtilityColumns(date_column, id_column, target_column, prediction_column),