[incubator-mxnet] branch master updated: Remove MXNET_STORAGE_FALLBACK_LOG_VERBOSE from test_autograd.py (#13830)

[marcoabreu]

This is an automated email from the ASF dual-hosted git repository.

marcoabreu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new abb7ce5  Remove MXNET_STORAGE_FALLBACK_LOG_VERBOSE from 
test_autograd.py (#13830)
abb7ce5 is described below

commit abb7ce528cf18fbc6bed29c920331212cf0fbb3a
Author: Pedro Larroy <pedro.larroy.li...@gmail.com>
AuthorDate: Fri Jan 11 01:08:24 2019 +0100

    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 2f88984..5b5aff3 100644
--- a/tests/python/unittest/test_autograd.py
+++ b/tests/python/unittest/test_autograd.py
@@ -21,6 +21,7 @@ from mxnet.ndarray import zeros_like
 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 test_unary_func():
         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 test_binary_func():
     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()