Update _an_attenuation.py

fmagrini · Sep 11, 2023 · da92e71 · da92e71
1 parent e50469f
commit da92e71
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/seislib/an/_an_attenuation.py b/seislib/an/_an_attenuation.py
@@ -753,7 +753,9 @@ def load_done_times():
                 if np.all(data_tmp == 0):
                     continue
                 fft = fourier_transform(data_tmp, window_samples)
-                if np.any(np.isnan(fft)):
+                if np.any(~np.isfinite(fft)):
+                    continue
+                if np.allclose(fft, 0):
                     continue
                 store.append(fft)
                 times_available.append(time)