Fix trapezoid call

StingraySoftware · Sep 2, 2024 · 0b197e8 · 0b197e8
1 parent 697831e
commit 0b197e8
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/stingray/multitaper.py b/stingray/multitaper.py
@@ -6,6 +6,12 @@
 import warnings
 
 import numpy as np
+
+try:
+    from numpy import trapezoid
+except ImportError:
+    from numpy import trapz as trapezoid
+
 import scipy.optimize
 import scipy.stats
 from scipy import signal, interpolate
@@ -460,7 +466,7 @@ def _get_adaptive_psd(self, freq_response, eigvals, max_iter=150):
 
         psd_est = self.psd_from_freq_response(freq_response, sqrt_eigvals[:, np.newaxis])
 
-        var = np.trapz(psd_est, dx=np.pi / n_freqs) / (2 * np.pi)
+        var = trapezoid(psd_est, dx=np.pi / n_freqs) / (2 * np.pi)
         del psd_est
 
         psd = np.empty(n_freqs)