diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py index 7aba53c97fc3..f25e77e30f9a 100644 --- a/tests/nightly/test_large_array.py +++ b/tests/nightly/test_large_array.py @@ -23,8 +23,9 @@ from mxnet.test_utils import rand_ndarray, assert_almost_equal, rand_coord_2d, default_context, check_symbolic_forward, create_2d_tensor from mxnet import gluon, nd -from tests.python.unittest.common import with_seed, with_post_test_cleanup +from tests.python.unittest.common import with_seed, with_post_test_cleanup, teardown from nose.tools import with_setup +import unittest # dimension constants MEDIUM_X = 10000 @@ -128,6 +129,7 @@ def np_softmax(x, axis=-1, temperature=1.0): x /= np.sum(x, axis=axis, keepdims=True) return x + @unittest.skip("log_softmax flaky, tracked at https://github.com/apache/incubator-mxnet/issues/17397") def check_log_softmax(): ndim = 2 shape = (SMALL_Y, LARGE_X) @@ -474,6 +476,7 @@ def check_ndarray_random_uniform(): a = nd.random.uniform(shape=(LARGE_X, SMALL_Y)) assert a[-1][0] != 0 + @unittest.skip("Randint flaky, tracked at https://github.com/apache/incubator-mxnet/issues/16172") @with_seed() def check_ndarray_random_randint(): a = nd.random.randint(100, 10000, shape=(LARGE_X, SMALL_Y)) @@ -1089,6 +1092,7 @@ def check_sort(): s = nd.sort(b, is_ascend=False) assert np.sum(s[0].asnumpy() == 0).all() + @unittest.skip("Topk takes lot of memory!, tracked at https://github.com/apache/incubator-mxnet/issues/17411") def check_topk(): b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y) k = nd.topk(b, k=10, axis=0, dtype=np.int64) diff --git a/tests/nightly/test_large_vector.py b/tests/nightly/test_large_vector.py index 6a55810613d4..f138467882c6 100644 --- a/tests/nightly/test_large_vector.py +++ b/tests/nightly/test_large_vector.py @@ -23,7 +23,7 @@ from mxnet.test_utils import rand_ndarray, assert_almost_equal, rand_coord_2d, create_vector from mxnet import gluon, nd -from tests.python.unittest.common import with_seed +from tests.python.unittest.common import with_seed, teardown from nose.tools import with_setup import unittest