Push the unadded changes

apache · May 10, 2020 · 8c626df · 8c626df
1 parent ec24155
commit 8c626df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_memory_opt.py b/tests/python/unittest/test_memory_opt.py
@@ -48,7 +48,7 @@ def _memory_opt_wrapper():
 
         if do_memory_opt:
             test_func()
-            del os.environ["MXNET_MEMORY_OPT"]
+            os.environ["MXNET_MEMORY_OPT"] = '0'
     return _memory_opt_wrapper