Fix test_ndarray

apache · Apr 5, 2019 · 1a5edfb · 1a5edfb
1 parent a6c1f0b
commit 1a5edfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_ndarray.py b/tests/python/unittest/test_ndarray.py
@@ -123,7 +123,7 @@ def test_ndarray_setitem():
     # numpy assignment for empty axis
     for trivial_shape in [(), (1,), (1, 1), (1, 1, 1)]:
         if trivial_shape == tuple():
-            with mx.numpy.enable_np_comp():
+            with mx.enable_np_comp():
                 x = mx.nd.zeros(trivial_shape)
         else:
             x = mx.nd.zeros(trivial_shape)