Skip to content

Commit d6b8fd2

Browse files
authored
Update summary stats return logic (TGSAI#716)
1 parent 69dcc2a commit d6b8fd2

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/mdio/segy/_workers.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,9 +158,15 @@ def trace_worker( # noqa: PLR0913
158158
data_array[region_slices] = tmp_samples
159159

160160
nonzero_samples = np.ma.masked_values(traces.sample, 0, copy=False)
161+
162+
nonzero_count = nonzero_samples.count()
163+
if nonzero_count == 0:
164+
# Return None to avoid calculating a NaN in sum_squares
165+
return None
166+
161167
histogram = CenteredBinHistogram(bin_centers=[], counts=[])
162168
return SummaryStatistics(
163-
count=nonzero_samples.count(),
169+
count=nonzero_count,
164170
min=nonzero_samples.min(),
165171
max=nonzero_samples.max(),
166172
sum=nonzero_samples.sum(dtype="float64"),

0 commit comments

Comments
 (0)