fix the test error

apache · Nov 10, 2019 · ac03a24 · ac03a24
1 parent 8e28e69
commit ac03a24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py
@@ -1763,8 +1763,8 @@ def hybrid_forward(self, F, a, b, *args, **kwargs):
 
     for lshape, rshape in shape_pairs:
         for func in funcs:
-            x1 = np.random.uniform(size=lshape) > 0.5
-            x2 = np.random.uniform(size=rshape) > 0.5
+            x1 = np.array(_np.random.uniform(size=lshape) > 0.5)
+            x2 = np.array(_np.random.uniform(size=rshape) > 0.5)
             check_boolean_binary_func(func, x1, x2)