@@ -199,9 +199,7 @@ def _gen_events(n_epochs):
199199 return events
200200
201201
202- def _reject_data_segments (
203- data , reject , flat , decim , info , tstep , * , return_drop_var = False
204- ):
202+ def _reject_data_segments (data , reject , flat , decim , info , tstep ):
205203 """Reject data segments using peak-to-peak amplitude."""
206204 from .._fiff .pick import channel_indices_by_type
207205 from ..epochs import _is_good
@@ -214,7 +212,6 @@ def _reject_data_segments(
214212 this_start = 0
215213 this_stop = 0
216214 drop_inds = []
217- drop_var = []
218215 for first in range (0 , data .shape [1 ], step ):
219216 last = first + step
220217 data_buffer = data [:, first :last ]
@@ -234,23 +231,14 @@ def _reject_data_segments(
234231 else :
235232 logger .info (f"Artifact detected in [{ first } , { last } ]" )
236233 drop_inds .append ((first , last ))
237- if return_drop_var :
238- drop_var .append (np .var (data_buffer , axis = 1 ))
239234 data = data_clean [:, :this_stop ]
240235 if not data .any ():
241236 raise RuntimeError (
242237 "No clean segment found. Please "
243238 "consider updating your rejection "
244239 "thresholds."
245240 )
246- out = (data , drop_inds )
247- if return_drop_var :
248- drop_var = np .reshape (
249- np .array (drop_var ), (len (drop_inds ), data .shape [0 ]), copy = False
250- )
251- assert drop_var .shape == (len (drop_inds ), data .shape [0 ])
252- out += (drop_var ,)
253- return out
241+ return data , drop_inds
254242
255243
256244def _get_inst_data (inst ):
0 commit comments