@@ -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