Fix use of np.trapezoid

openPMD · Oct 23, 2024 · d5f79b4 · d5f79b4
1 parent 6caaff6
commit d5f79b4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/openpmd_viewer/addons/pic/lpa_diagnostics.py b/openpmd_viewer/addons/pic/lpa_diagnostics.py
@@ -1041,7 +1041,12 @@ def get_spectrogram( self, t=None, iteration=None, pol=None,
         T = tmax - tmin
         dt = T / Nz
         # Normalize the Envelope
-        env /= np.sqrt(np.trapezoid(env ** 2, dx=dt))
+        # Note that np.trapz has been deprecated and is replaced by np.trapezoid in numpy2
+        try:
+            trapezoid = np.trapezoid
+        except AttributeError:
+            trapezoid = np.trapz
+        env /= np.sqrt(trapezoid(env ** 2, dx=dt))
         # Allocate array for the gating function and the spectrogran
         E_shift = np.zeros_like(E)
         spectrogram = np.zeros((2 * Nz, Nz))