Skip to content

Commit 9464532

Browse files
committed
small fix to avoid dask median NaN issues
1 parent b2de417 commit 9464532

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

oceanstream/echodata/denoise/impulse_noise.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,9 @@ def impulse_noise_mask(
172172
range_values = range_values.compute()
173173
elif range_values.dims:
174174
range_dim = range_values.dims[0]
175+
# Materialise to avoid dask median NaN issues
176+
if hasattr(range_values, "compute"):
177+
range_values = range_values.compute()
175178
else:
176179
range_dim = range_coord
177180

@@ -184,6 +187,8 @@ def impulse_noise_mask(
184187
window_m = float(s.rstrip("mM "))
185188
dz_arr = range_values.diff(range_dim)
186189
dz = float(dz_arr.median()) if dz_arr.size > 0 else 0.1
190+
if np.isnan(dz) or dz == 0:
191+
dz = 0.1
187192
bin_sz = max(1, int(round(window_m / abs(dz))))
188193
else:
189194
bin_sz = max(1, int(s))

0 commit comments

Comments
 (0)