For comparison, tensors need to be converted to numpy format

because PR apache#14443 is not approved yet in upstream
drivanov · Sep 27, 2019 · 6d86117 · 6d86117
1 parent 8358546
commit 6d86117
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py
@@ -7625,7 +7625,7 @@ def run_test_argmax(multi_output, ctx, axis=None):
 
         max = mx.nd.argmax(tensor, axis=axis)
         topk_data = mx.nd.topk(tensor, axis=axis, is_ascend=0, k=1)
-        assert_almost_equal(max.reshape(-1), topk_data.reshape(-1))
+        assert_almost_equal(max.reshape(-1).asnumpy(), topk_data.reshape(-1).asnumpy())
 
     ctx = default_context()
     for multi_output in [False, True]: