Fix np.choice

apache · Sep 18, 2019 · 477e6f7 · 477e6f7
1 parent 681d9a7
commit 477e6f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py
@@ -1723,7 +1723,7 @@ def test_indexing_mode(sampler, set_size, samples_size, replace, weight=None):
             samples = sampler(a)
         assert len(samples) == samples_size
         if not replace:
-            assert len(_np.unique(samples)) == samples_size
+            assert len(_np.unique(samples.asnumpy())) == samples_size
 
     num_classes = 10
     num_samples = 10 ** 8