Import AxisError from dpctl.tensor._numpy_helper

Author: oleksandr-pavlyk

dpctl/tests/test_usm_ndarray_utility_functions.py

@@ -2,10 +2,10 @@
 
 import numpy as np
 import pytest
-from numpy import AxisError
 from numpy.testing import assert_array_equal, assert_equal
 
 import dpctl.tensor as dpt
+from dpctl.tensor._numpy_helper import AxisError
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
 _all_dtypes = [