Skip to content

Commit ae5dedc

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent de6177c commit ae5dedc

1 file changed

Lines changed: 13 additions & 11 deletions

File tree

src/spikeinterface/extractors/neoextractors/neobaseextractor.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -231,22 +231,24 @@ def __init__(
231231

232232
# remove all duplicate channel-to-electrode assignments
233233
mask_id = np.argwhere(mask).flatten()
234-
[u,u_i,u_v,u_c] = np.unique(signal_channels[mask]['name'], return_index=True, return_inverse=True, return_counts=True)
235-
for i in u_v[u_i[u_c>1]]:
236-
mask[mask_id[np.argwhere(signal_channels[mask]['name'] == u[i])[1:].flatten()]] = False
237-
234+
[u, u_i, u_v, u_c] = np.unique(
235+
signal_channels[mask]["name"], return_index=True, return_inverse=True, return_counts=True
236+
)
237+
for i in u_v[u_i[u_c > 1]]:
238+
mask[mask_id[np.argwhere(signal_channels[mask]["name"] == u[i])[1:].flatten()]] = False
239+
238240
# remove all duplicate channel assigments corresponding to different electrodes (channel is a mix of mulitple electrode signals)
239241
mask_id = np.argwhere(mask).flatten()
240-
signal_channels_chan,_ = map(list, zip(*(x.split(' ') for x in signal_channels[mask]['name'])))
241-
[u,u_i,u_v,u_c] = np.unique(signal_channels_chan, return_index=True, return_inverse=True, return_counts=True)
242-
for i in u_v[u_i[u_c>1]]:
242+
signal_channels_chan, _ = map(list, zip(*(x.split(" ") for x in signal_channels[mask]["name"])))
243+
[u, u_i, u_v, u_c] = np.unique(signal_channels_chan, return_index=True, return_inverse=True, return_counts=True)
244+
for i in u_v[u_i[u_c > 1]]:
243245
mask[mask_id[np.argwhere(signal_channels_chan == u[i])[:].flatten()]] = False
244-
246+
245247
# remove subsequent duplicated electrodes (single electrode saved to multiple channels)
246248
mask_id = np.argwhere(mask).flatten()
247-
_,signal_channels_elec = map(list, zip(*(x.split(' ') for x in signal_channels[mask]['name'])))
248-
[u,u_i,u_v,u_c] = np.unique(signal_channels_elec, return_index=True, return_inverse=True, return_counts=True)
249-
for i in u_v[u_i[u_c>1]]:
249+
_, signal_channels_elec = map(list, zip(*(x.split(" ") for x in signal_channels[mask]["name"])))
250+
[u, u_i, u_v, u_c] = np.unique(signal_channels_elec, return_index=True, return_inverse=True, return_counts=True)
251+
for i in u_v[u_i[u_c > 1]]:
250252
mask[mask_id[np.argwhere(signal_channels_elec == u[i])[1:].flatten()]] = False
251253

252254
signal_channels = signal_channels[mask]

0 commit comments

Comments
 (0)