diff --git a/ceilopyter/noise.py b/ceilopyter/noise.py index cba329c..a7a26a7 100644 --- a/ceilopyter/noise.py +++ b/ceilopyter/noise.py @@ -5,7 +5,7 @@ def remove_noise( beta_uncorr: npt.NDArray[np.floating], noise_floor: float, snr_limit: float = 5 -) -> npt.NDArray[np.bool]: +) -> npt.NDArray[np.bool_]: zero_ranges = np.all(beta_uncorr == 0, axis=0) n_zeros = np.argmax(~zero_ranges[::-1])