Skip to content

Commit 95537fe

Browse files
authored
Merge pull request #1079 from alejoe91/small_fixed
Fixes from SpikeInterface CI
2 parents f7792e5 + f49c649 commit 95537fe

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

neo/rawio/axonarawio.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -304,6 +304,9 @@ def _get_analogsignal_chunk(self, block_index, seg_index, i_start, i_stop,
304304
i_stop = bin_dict['num_total_samples']
305305
if channel_indexes is None:
306306
channel_indexes = [i for i in range(bin_dict['num_channels'])]
307+
elif isinstance(channel_indexes, slice):
308+
channel_indexes_all = [i for i in range(bin_dict['num_channels'])]
309+
channel_indexes = channel_indexes_all[channel_indexes]
307310

308311
num_samples = (i_stop - i_start)
309312

neo/rawio/biocamrawio.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,9 +112,10 @@ def _get_analogsignal_chunk(self, block_index, seg_index, i_start, i_stop,
112112
i_start = 0
113113
if i_stop is None:
114114
i_stop = self._num_frames
115-
115+
if channel_indexes is None:
116+
channel_indexes = slice(None)
116117
data = self._read_function(self._filehandle, i_start, i_stop, self._num_channels)
117-
return np.squeeze(data[:, channel_indexes])
118+
return data[:, channel_indexes]
118119

119120

120121
def open_biocam_file_header(filename):

0 commit comments

Comments
 (0)