From abb7ce528cf18fbc6bed29c920331212cf0fbb3a Mon Sep 17 00:00:00 2001 From: Pedro Larroy Date: Fri, 11 Jan 2019 01:08:24 +0100 Subject: [PATCH] Remove MXNET_STORAGE_FALLBACK_LOG_VERBOSE from test_autograd.py (#13830) --- tests/python/unittest/test_autograd.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/tests/python/unittest/test_autograd.py b/tests/python/unittest/test_autograd.py index 2f889845af3f..5b5aff31231b 100644 --- a/tests/python/unittest/test_autograd.py +++ b/tests/python/unittest/test_autograd.py @@ -21,6 +21,7 @@ from mxnet.autograd import * from mxnet.test_utils import * from common import setup_module, with_seed, teardown +from mxnet.test_utils import EnvManager def grad_and_loss(func, argnum=None): @@ -120,8 +121,9 @@ def check_unary_func(x): autograd_assert(x, func=f_square, grad_func=f_square_grad) uniform = nd.uniform(shape=(4, 5)) stypes = ['default', 'row_sparse', 'csr'] - for stype in stypes: - check_unary_func(uniform.tostype(stype)) + with EnvManager('MXNET_STORAGE_FALLBACK_LOG_VERBOSE', '0'): + for stype in stypes: + check_unary_func(uniform.tostype(stype)) @with_seed() def test_binary_func(): @@ -138,11 +140,12 @@ def check_binary_func(x, y): uniform_x = nd.uniform(shape=(4, 5)) uniform_y = nd.uniform(shape=(4, 5)) stypes = ['default', 'row_sparse', 'csr'] - for stype_x in stypes: - for stype_y in stypes: - x = uniform_x.tostype(stype_x) - y = uniform_y.tostype(stype_y) - check_binary_func(x, y) + with EnvManager('MXNET_STORAGE_FALLBACK_LOG_VERBOSE', '0'): + for stype_x in stypes: + for stype_y in stypes: + x = uniform_x.tostype(stype_x) + y = uniform_y.tostype(stype_y) + check_binary_func(x, y) @with_seed()