fix CI error

apache · May 12, 2020 · e7f0f36 · e7f0f36
1 parent 357be34
commit e7f0f36
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
@@ -2388,7 +2388,7 @@ def hybrid_forward(self, F, a, *args, **kwargs):
         np_func = getattr(_np, func)
         mx_func = TestUnary(func)
         np_test_data = _np.random.uniform(low, high, shape).astype(_np.int32)
-        mx_test_data = mx.numpy.array(np_test_data)
+        mx_test_data = mx.numpy.array(np_test_data).astype(_np.int32)
         for hybridize in [True, False]:
             if hybridize:
                 mx_func.hybridize()