Fix type check to include numpy integer types

Signed-off-by: abigailt <[email protected]>

Author: abigailgold

art/estimators/classification/classifier.py

@@ -114,7 +114,7 @@ def nb_classes(self, nb_classes: int):
         """
         Set the number of output classes.
         """
-        if nb_classes is None or (isinstance(nb_classes, int) and nb_classes < 2):
+        if nb_classes is None or (isinstance(nb_classes, (int, np.integer)) and nb_classes < 2):
             raise ValueError("nb_classes must be greater than or equal to 2.")
 
         self._nb_classes = nb_classes

art/utils.py

@@ -801,7 +801,7 @@ def check_and_transform_label_format(
     """
     labels_return = labels
 
-    if nb_classes is not None and not isinstance(nb_classes, int):
+    if nb_classes is not None and not isinstance(nb_classes, (int, np.integer)):
         raise TypeError("nb_classes that is not an integer is not supported")
 
     if len(labels.shape) == 2 and labels.shape[1] > 1:  # multi-class, one-hot encoded