Merge pull request #786 from yonishelach/feature-selection-np-object

[Feature-selection] Update numpy object use (deprecated)
mlrun · Feb 4, 2024 · 438128d · 438128d
2 parents a7a0517 + cf04a89
commit 438128d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 18 deletions.
diff --git a/feature_selection/feature_selection.py b/feature_selection/feature_selection.py
@@ -180,7 +180,7 @@ def feature_selection(
     y = df.pop(label_column)
     X = df
 
-    if np.object in list(X.dtypes) and ignore_type_errors is False:
+    if np.object_ in list(X.dtypes) and ignore_type_errors is False:
         raise ValueError(
             f"{df.select_dtypes(include=['object']).columns.tolist()} are neither float or int."
         )