@@ -412,7 +412,7 @@ def _parse_header(self):
412412 for k , (data , ev_ids ) in self .nev_data .items ():
413413 segment_mask = ev_ids == data_bl
414414 if data [segment_mask ].size > 0 :
415- t = (data [segment_mask ][- 1 ]['timestamp' ]+ 1 ) / self .__nev_basic_header [
415+ t = (data [segment_mask ][- 1 ]['timestamp' ] + 1 ) / self .__nev_basic_header [
416416 'timestamp_resolution' ]
417417 max_nev_time = max (max_nev_time , t )
418418 if max_nev_time > t_stop :
@@ -450,7 +450,8 @@ def _parse_header(self):
450450 # Calculate t_start and t_stop for each segment in seconds
451451 resolution = self .__nev_basic_header ['timestamp_resolution' ]
452452 self ._seg_t_starts = [v / float (resolution ) for k , v in sorted (min_nev_times .items ())]
453- self ._seg_t_stops = [(v + 1 ) / float (resolution ) for k , v in sorted (max_nev_times .items ())]
453+ self ._seg_t_stops = [(v + 1 ) / float (resolution )
454+ for k , v in sorted (max_nev_times .items ())]
454455 self ._nb_segment = len (self ._seg_t_starts )
455456 self ._sigs_t_starts = [None ] * self ._nb_segment
456457
@@ -648,7 +649,7 @@ def _get_spike_raw_waveforms(self, block_index, seg_index, unit_index, t_start,
648649 wf_byte_size = np .dtype (wf_dtype ).itemsize * wf_size
649650
650651 dt1 = [
651- ('extra' , 'S{}' .format (unit_spikes ['waveform' ].dtype .itemsize - wf_byte_size )),
652+ ('extra' , 'S{}' .format (unit_spikes ['waveform' ].dtype .itemsize - wf_byte_size )),
652653 ('ch_waveform' , 'S{}' .format (wf_byte_size ))]
653654
654655 waveforms = unit_spikes ['waveform' ].view (dt1 )['ch_waveform' ].flatten ().view (wf_dtype )
0 commit comments