BUG: (NEP 19) fix array repr compatibility with Numpy 2.0

unyt/_array_functions.py

@@ -995,3 +995,14 @@ def interp(x, xp, fp, *args, **kwargs):
         np.interp(np.asarray(x), np.asarray(xp), np.asarray(fp), *args, **kwargs)
         * ret_units
     )
+
+
+@implements(np.array_repr)
+def array_repr(arr, *args, **kwargs):
+    rep = np.array_repr._implementation(arr.view(np.ndarray), *args, **kwargs)
+    rep = rep.replace("array", arr.__class__.__name__)
+    units_repr = arr.units.__repr__()
+    if "=" in rep:
+        return rep[:-1] + ", units='" + units_repr + "')"
+    else:
+        return rep[:-1] + ", '" + units_repr + "')"

unyt/array.py

@@ -636,12 +636,7 @@ def __new__(
         return obj
 
     def __repr__(self):
-        rep = super().__repr__()
-        units_repr = self.units.__repr__()
-        if "=" in rep:
-            return rep[:-1] + ", units='" + units_repr + "')"
-        else:
-            return rep[:-1] + ", '" + units_repr + "')"
+        return np.array_repr(self)
 
     def __str__(self):
         return str(self.view(np.ndarray)) + " " + str(self.units)

unyt/tests/test_array_functions.py

@@ -35,7 +35,6 @@
     np.argpartition,  # returns pure numbers
     np.argsort,  # returns pure numbers
     np.argwhere,  # returns pure numbers
-    np.array_repr,  # hooks into __repr__
     np.array_str,  # hooks into __str__
     np.atleast_1d,  # works out of the box (tested)
     np.atleast_2d,  # works out of the box (tested)
@@ -256,7 +255,7 @@ def test_wrapping_completeness():
 
 def test_array_repr():
     arr = [1, 2, 3] * cm
-    assert np.array_repr(arr) == "unyt_array([1, 2, 3], units='cm')"
+    assert np.array_repr(arr) == "unyt_array([1, 2, 3], 'cm')"
 
 
 def test_dot_vectors():