diff --git a/tests/python/unittest/test_numpy_ndarray.py b/tests/python/unittest/test_numpy_ndarray.py index db8b18b49bf2..38798a545f3e 100644 --- a/tests/python/unittest/test_numpy_ndarray.py +++ b/tests/python/unittest/test_numpy_ndarray.py @@ -19,6 +19,7 @@ from __future__ import absolute_import from __future__ import division import os +import unittest import numpy as _np import mxnet as mx from mxnet import np, npx, autograd @@ -977,6 +978,7 @@ def test_np_multinomial(): @with_seed() +@unittest.skipUnless(has_tvm_ops(), "Comparison ops are implemented using TVM") @use_np def test_np_ndarray_boolean_indexing(): def test_single_bool_index():