Merge pull request #19507 from neutrinoceros/compat/support-np2_5-ufuncs

mhvk · web-flow · commit e05b248dd613 · 2026-04-02T14:46:14.000+02:00
BUG: add support for new internal ufuncs in numpy 2.5 (`np._core.imag` and `np._core.real`)
diff --git a/astropy/units/quantity_helper/helpers.py b/astropy/units/quantity_helper/helpers.py
@@ -18,6 +18,7 @@
     NUMPY_LT_2_1,
     NUMPY_LT_2_2,
     NUMPY_LT_2_3,
+    NUMPY_LT_2_5,
 )
 
 from . import UFUNC_HELPERS, UNSUPPORTED_UFUNCS
@@ -454,6 +455,9 @@ def helper_clip(f, unit1, unit2, unit3):
     np.trunc,
     np.positive,
 )
+if not NUMPY_LT_2_5:
+    invariant_ufuncs += (np_umath.imag, np_umath.real)
+
 for ufunc in invariant_ufuncs:
     UFUNC_HELPERS[ufunc] = helper_invariant
 
