Skip to content

Commit 7bf0cee

Browse files
committed
FIX: Restore
1 parent 5e0c34c commit 7bf0cee

2 files changed

Lines changed: 3 additions & 14 deletions

File tree

mne/_fiff/pick.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1309,6 +1309,7 @@ def _picks_str_to_idx(
13091309
#
13101310
# first: check our special cases
13111311
#
1312+
13121313
picks_generic = list()
13131314
if len(picks) == 1:
13141315
if picks[0] in ("all", "data", "data_or_ica"):

mne/utils/numerics.py

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -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

256244
def _get_inst_data(inst):

0 commit comments

Comments
 (0)