[MON-2260] 'un_numpy' func fix (#2586)

* fix * fix * fix
deepchecks · Jun 8, 2023 · 8cf5507 · 8cf5507
1 parent 0cda40a
commit 8cf5507
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/deepchecks/utils/dataframes.py b/deepchecks/utils/dataframes.py
@@ -104,6 +104,14 @@ def un_numpy(val):
     -------
         returns the numpy value in a native type.
     """
+    if isinstance(val, np.str_):
+        # NOTE:
+        # 'np.str_' is instance of the 'np.generic' but
+        # 'np.isnan(np.str_())' raises an error with a next message:
+        # >> TypeError: ufunc 'isnan' not supported for the input types...)
+        #
+        # therefore this 'if' statement is needed
+        return val.item()
     if isinstance(val, np.generic):
         if np.isnan(val):
             return None

diff --git a/spelling-allowlist.txt b/spelling-allowlist.txt
@@ -159,3 +159,6 @@ huggingface
 tiktoken
 ai
 idx
+isnan
+'isnan'
+ufunc