Change check for nullness of arrays

credo-ai · Jan 17, 2023 · 342319c · 342319c
1 parent f41e460
commit 342319c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/credoai/artifacts/model/classification_model.py b/credoai/artifacts/model/classification_model.py
@@ -163,11 +163,11 @@ def __init__(
         tags=None,
     ):
         self.model_like = model_like
-        if predict_output:
+        if predict_output is not None:
             self.predict_output = check_array(
                 predict_output, ensure_2d=False, allow_nd=True
             )
-        if predict_proba_output:
+        if predict_proba_output is not None:
             self.predict_proba_output = check_array(
                 predict_proba_output, ensure_2d=False, allow_nd=True
             )