diff --git a/src/spikeinterface/preprocessing/remove_artifacts.py b/src/spikeinterface/preprocessing/remove_artifacts.py index 60780856b0..6eece4ce61 100644 --- a/src/spikeinterface/preprocessing/remove_artifacts.py +++ b/src/spikeinterface/preprocessing/remove_artifacts.py @@ -263,12 +263,14 @@ def get_traces(self, start_frame, end_frame, channel_indices): traces = self.parent_recording_segment.get_traces(start_frame, end_frame, channel_indices) traces = traces.copy() - mask = (self.triggers >= start_frame) & (self.triggers < end_frame) + pad = self.pad + if pad is None: + mask = (self.triggers >= start_frame) & (self.triggers < end_frame) + else: + mask = (self.triggers >= start_frame - pad[1]) & (self.triggers < end_frame + pad[0]) triggers = self.triggers[mask] - start_frame labels = self.labels[mask] - pad = self.pad - if self.mode == "zeros": for trig in triggers: if pad is None: