|
28 | 28 | from tqdm import tqdm, trange |
29 | 29 |
|
30 | 30 | from .baserawio import ( |
31 | | - BaseRawIO, _signal_channel_dtype, _signal_stream_dtype, _spike_channel_dtype, _event_channel_dtype |
| 31 | + BaseRawIO, |
| 32 | + _signal_channel_dtype, |
| 33 | + _signal_stream_dtype, |
| 34 | + _spike_channel_dtype, |
| 35 | + _event_channel_dtype, |
32 | 36 | ) |
33 | 37 |
|
34 | 38 |
|
@@ -69,7 +73,11 @@ def _parse_header(self): |
69 | 73 | nb_event_chan = global_header['NumEventChannels'] |
70 | 74 | offset2 = offset1 + np.dtype(DspChannelHeader).itemsize * nb_unit_chan |
71 | 75 | eventHeaders = np.memmap( |
72 | | - self.filename, dtype=EventChannelHeader, mode='r', offset=offset2, shape=(nb_event_chan,) |
| 76 | + self.filename, |
| 77 | + dtype=EventChannelHeader, |
| 78 | + mode='r', |
| 79 | + offset=offset2, |
| 80 | + shape=(nb_event_chan,), |
73 | 81 | ) |
74 | 82 |
|
75 | 83 | # slow channel header = signal |
@@ -123,7 +131,11 @@ def _parse_header(self): |
123 | 131 | } |
124 | 132 | for bl_type in tqdm(block_pos, desc="Finalizing data blocks", leave=True): |
125 | 133 | self._data_blocks[bl_type] = {} |
126 | | - for chan_id in tqdm(block_pos[bl_type], desc="Finalizing data blocks for type %d" % bl_type, leave=True): |
| 134 | + for chan_id in tqdm( |
| 135 | + block_pos[bl_type], |
| 136 | + desc="Finalizing data blocks for type %d" % bl_type, |
| 137 | + leave=True, |
| 138 | + ): |
127 | 139 | positions = block_pos[bl_type][chan_id] |
128 | 140 | dt = dtype_by_bltype[bl_type] |
129 | 141 | data_block = np.empty((len(positions)), dtype=dt) |
@@ -220,7 +232,11 @@ def _parse_header(self): |
220 | 232 |
|
221 | 233 | # Spikes channels |
222 | 234 | spike_channels = [] |
223 | | - for unit_index, (chan_id, unit_id) in tqdm(enumerate(self.internal_unit_ids), desc="Parsing spike channels", leave=True): |
| 235 | + for unit_index, (chan_id, unit_id) in tqdm( |
| 236 | + enumerate(self.internal_unit_ids), |
| 237 | + desc="Parsing spike channels", |
| 238 | + leave=True, |
| 239 | + ): |
224 | 240 | c = np.nonzero(dspChannelHeaders['Channel'] == chan_id)[0][0] |
225 | 241 | h = dspChannelHeaders[c] |
226 | 242 |
|
|
0 commit comments