wip agu 2024, small bugfix

ROBelgium · Dec 6, 2023 · e7cf414 · e7cf414
1 parent 4c8f23a
commit e7cf414
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/msnoise/move2obspy.py b/msnoise/move2obspy.py
@@ -283,7 +283,7 @@ def whiten2(fft, Nfft, low, high, porte1, porte2, psds, whiten_type):
             fft[i, 0:low] *= 0
             fft[i, high:] *= 0
         elif whiten_type == "HANN":
-            fft[i] /= np.abs(fft[i])
+            fft[i] = np.divide(fft[i], np.abs(fft[i]), out=np.zeros_like(fft[i]), where=np.abs(fft[i])!=0)
             fft[i][:porte1] *= 0.0
             fft[i][porte1:porte2 + 1] *= hann
             fft[i][porte2 + 1:] *= 0.0