Fix numpy 2 errors

PiperOrigin-RevId: 665992136

Author: Conchylicultor

dataclass_array/array_dataclass_test.py

@@ -520,13 +520,13 @@ def test_absolute_axis(xnp: enp.NpModule):
   assert p._to_absolute_axis((0, 1, -1)) == (0, 1, 3)
   assert p._to_absolute_axis((-1, -2)) == (3, 2)
 
-  with pytest.raises(np.AxisError):
+  with pytest.raises(np.exceptions.AxisError):
     assert p._to_absolute_axis(4)
 
-  with pytest.raises(np.AxisError):
+  with pytest.raises(np.exceptions.AxisError):
     assert p._to_absolute_axis(-5)
 
-  with pytest.raises(np.AxisError):
+  with pytest.raises(np.exceptions.AxisError):
     assert p._to_absolute_axis((0, 4))
 
 

dataclass_array/utils/np_utils.py

@@ -157,7 +157,7 @@ def to_absolute_axis(axis: Axes, *, ndim: int) -> Axes:
     return tuple(range(ndim))
   elif isinstance(axis, int):
     if axis >= ndim or axis < -ndim:
-      raise enp.lazy.np.AxisError(
+      raise enp.lazy.np.exceptions.AxisError(
           axis=axis,
           ndim=ndim,
           # msg_prefix=

dataclass_array/utils/np_utils_test.py

@@ -55,13 +55,13 @@ def test_to_absolute_axis():
   assert np_utils.to_absolute_axis((0, 1, -1), ndim=4) == (0, 1, 3)
   assert np_utils.to_absolute_axis((-1, -2), ndim=4) == (3, 2)
 
-  with pytest.raises(np.AxisError):
+  with pytest.raises(np.exceptions.AxisError):
     assert np_utils.to_absolute_axis(4, ndim=4)
 
-  with pytest.raises(np.AxisError):
+  with pytest.raises(np.exceptions.AxisError):
     assert np_utils.to_absolute_axis(-5, ndim=4)
 
-  with pytest.raises(np.AxisError):
+  with pytest.raises(np.exceptions.AxisError):
     assert np_utils.to_absolute_axis((0, 4), ndim=4)