diff --git a/tests/python/gpu/test_operator_gpu.py b/tests/python/gpu/test_operator_gpu.py index 6af9a0f33d48..bb18d5d21c87 100644 --- a/tests/python/gpu/test_operator_gpu.py +++ b/tests/python/gpu/test_operator_gpu.py @@ -33,7 +33,7 @@ from test_gluon_rnn import * from test_sparse_operator import test_cast_storage_ex, test_sparse_dot from test_sparse_operator import test_sparse_nd_zeros, test_sparse_retain -from test_sparse_ndarray import test_create_csr, test_create_row_sparse +from test_sparse_ndarray import test_create_csr, test_create_row_sparse, test_sparse_nd_setitem set_default_context(mx.gpu(0)) del test_support_vector_machine_l1_svm