diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py index 07aa37b1de66..2dcc9f3c9e6d 100644 --- a/tests/python/unittest/test_numpy_op.py +++ b/tests/python/unittest/test_numpy_op.py @@ -2728,9 +2728,9 @@ def GetNdim(tp): obj_onp = obj test_insert = TestInsert(obj=obj_mxnp, axis=axis) - a = mx.nd.random.uniform(-1.0, 1.0, shape=arr_shape).as_np_ndarray().astype(atype) + a = mx.nd.random.uniform(-10.0, 10.0, shape=arr_shape).as_np_ndarray().astype(atype) a.attach_grad() - b = mx.nd.random.uniform(-1.0, 1.0, shape=val_shape).as_np_ndarray().astype(btype) + b = mx.nd.random.uniform(-10.0, 10.0, shape=val_shape).as_np_ndarray().astype(btype) b.attach_grad() expected_ret = _np.insert(a.asnumpy(), obj_onp, b.asnumpy(), axis=axis) with mx.autograd.record():