diff --git a/src/ess/nmx/mcstas/load.py b/src/ess/nmx/mcstas/load.py index b989d09..186e671 100644 --- a/src/ess/nmx/mcstas/load.py +++ b/src/ess/nmx/mcstas/load.py @@ -49,13 +49,10 @@ def load_event_data_bank_name( def _exclude_zero_events(data: sc.Variable) -> sc.Variable: """Exclude events with zero counts from the data. - McStas can add an extra event line containing 0,0,0,0,0,0 - This line should not be included so we skip it. + McStas can add extra event lines containing 0,0,0,0,0,0 + These lines should not be included so we skip it. """ - if (data.values[0] == 0).all(): - data = data["event", 1:] - else: - data = data + data = data[(data != sc.scalar(0.0, unit=data.unit)).any(dim="dim_1")] return data