global trap function

d-chambers · d-chambers · commit 16f8db8e7a33 · 2025-12-22T09:52:33.000-07:00
diff --git a/dascore/transform/integrate.py b/dascore/transform/integrate.py
@@ -16,6 +16,8 @@
     patch_function,
 )
 
+_TRAP_FUNC = getattr(np, "trapezoid" if hasattr(np, "trapezoid") else "trapz")
+
 
 def _quasi_mean(array):
     """Get a quasi mean value from an array. Works with datetimes."""
@@ -45,13 +47,11 @@ def _get_new_coords_and_array(patch, array, dims):
     ndims = len(patch.shape)
     for dxs_or_val, ax in zip(dxs_or_vals, axes):
         # Numpy 2/3 compat code
-        trap_name = "trapezoid" if hasattr(np, "trapezoid") else "trapz"
-        trap = getattr(np, trap_name)
         indexer = broadcast_for_index(ndims, ax, None, fill=slice(None))
         if is_array(dxs_or_val):
-            array = trap(array, x=dxs_or_val, axis=ax)[indexer]
+            array = _TRAP_FUNC(array, x=dxs_or_val, axis=ax)[indexer]
         else:
-            array = trap(array, dx=dxs_or_val, axis=ax)[indexer]
+            array = _TRAP_FUNC(array, dx=dxs_or_val, axis=ax)[indexer]
     array, coords = _get_new_coords_and_array(patch, array, dims)
     return array, coords
 
