iter

Author: glemaitre

imblearn/dask/tests/test_utils.py

@@ -1,5 +1,7 @@
 import numpy as np
 import pytest
+
+dask = pytest.importorskip("dask")
 from dask import array
 
 from imblearn.dask.utils import is_multilabel
@@ -18,7 +20,7 @@ def test_type_of_target_error():
     "y, expected_result",
     [
         (array.from_array(np.array([0, 1, 0, 1])), False),
-        (array.from_array(np.array([[1, 0], [0, 0]])), True),
+        (array(np.array([[1, 0], [0, 0]])), True),
         (array.from_array(np.array([[1], [0], [0]])), False),
         (array.from_array(np.array([[1, 0, 0]])), True),
     ]