Fix ci

apache · Aug 7, 2019 · 4234412 · 4234412
1 parent f4d2af0
commit 4234412
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py
@@ -4793,7 +4793,7 @@ def test_normal_case():
 
     def test_empty_tensor():
         shape = (2, 3, 0, 4)
-        with mx.np_compat():
+        with mx.np_shape():
             a = np.array([], dtype=np.int32).reshape(shape)
             b = mx.nd.array(a, ctx=default_context(), dtype=a.dtype)
 
@@ -4890,7 +4890,7 @@ def test_normal_case(index_type=np.int32):
 
     def test_empty_indices():
         shape = (2, 0, 9, 3)
-        with mx.np_compat():
+        with mx.np_shape():
             indices = np.array([]).reshape(shape)
             depth = 10
             mx_one_hot_array = mx.nd.one_hot(