Fix logic in binary indicator matrices check

drivendataorg · Nov 12, 2018 · 02d7650 · 02d7650
1 parent c7abc6f
commit 02d7650
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/data/multilabel.py b/src/data/multilabel.py
@@ -12,7 +12,7 @@ def multilabel_sample(y, size=1000, min_count=5, seed=None):
         each label.
     """
     try:
-        if (np.unique(y).astype(int) != np.array([0, 1])).all():
+        if (np.unique(y).astype(int) != np.array([0, 1])).any():
             raise ValueError()
     except (TypeError, ValueError):
         raise ValueError('multilabel_sample only works with binary indicator matrices')