task/Numpy encoder for classification_labels

python-client/giskard/models/base/__init__.py

@@ -24,6 +24,7 @@
 from giskard.ml_worker.utils.logging import Timer
 from giskard.path_utils import get_size
 from giskard.settings import settings
+from ..utils import np_types_to_native
 
 MODEL_CLASS_PKL = "ModelClass.pkl"
 
@@ -111,7 +112,7 @@ def __init__(
             name=name if name is not None else self.__class__.__name__,
             model_type=model_type,
             feature_names=list(feature_names) if feature_names else None,
-            classification_labels=classification_labels,
+            classification_labels=np_types_to_native(classification_labels),
             loader_class=self.__class__.__name__,
             loader_module=self.__module__,
             classification_threshold=classification_threshold,

python-client/giskard/models/utils.py

@@ -1,5 +1,7 @@
 import itertools
 from collections.abc import Iterator
+from typing import List
+import numpy as np
 
 
 def map_to_tuples(data: Iterator):
@@ -10,3 +12,9 @@ def map_to_tuples(data: Iterator):
         return data
 
     return map(lambda x: (x,), data)
+
+
+def np_types_to_native(some_list: List):
+    if some_list is not None:
+        return [i.item() if isinstance(i, np.generic) else i for i in some_list]
+    return some_list