diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py index 3b11b35742b1..27969c75f683 100644 --- a/tests/python/unittest/test_numpy_op.py +++ b/tests/python/unittest/test_numpy_op.py @@ -4382,6 +4382,7 @@ def test_np_randn(): @with_seed() @use_np +@pytest.mark.skip(reason='Test hangs. Tracked in #18144') def test_np_multivariate_normal(): class TestMultivariateNormal(HybridBlock): def __init__(self, size=None): @@ -8178,6 +8179,7 @@ def g(data): @with_seed() @use_np +@pytest.mark.skip(reason='Test hangs. Tracked in #18144') def test_np_resize(): class TestResize(HybridBlock): def __init__(self, new_shape): @@ -8514,13 +8516,13 @@ def hybrid_forward(self, F, a): mx_out_imperative = getattr(mx.np, func)(mx_data) assert_almost_equal(mx_out_imperative.asnumpy(), np_out, rtol, atol) # if `out` is given and dtype == np.bool - mx_x = np.empty_like(mx_data).astype(np.bool) + mx_x = np.ones_like(mx_data).astype(np.bool) np_x = mx_x.asnumpy() getattr(mx.np, func)(mx_data, mx_x) np_func(np_data, np_x) assert_almost_equal(mx_out_imperative .asnumpy(), np_out, rtol, atol) # if `out` is given but dtype mismatches - mx_y = np.empty_like(mx_data) + mx_y = np.ones_like(mx_data) assertRaises(TypeError, getattr(np, func), mx_data, out=mx_y) assertRaises(NotImplementedError, getattr(np, func), mx_data, where=False) @@ -8742,6 +8744,7 @@ def hybrid_forward(self, F, x): @with_seed() @use_np +@pytest.mark.skip(reason='Test hangs. Tracked in #18144') def test_np_unravel_index(): class TestUnravel_index(HybridBlock): def __init__(self, shape, order='C') :