Skip to content

Commit 28fdcec

Browse files
committed
eliminate unecessary loop
1 parent b1d7a8d commit 28fdcec

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

neo/rawio/intanrawio.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -185,15 +185,15 @@ def _parse_header(self):
185185
# based on https://github.com/NeuralEnsemble/python-neo/issues/1556 bug in versions 0.13.1, .2, .3
186186
elif self.file_format == "one-file-per-signal":
187187
self._raw_data = {}
188-
for stream_index, (stream_name, stream_datatype) in enumerate(memmap_data_dtype.items()):
188+
for stream_name, stream_dtype in memmap_data_dtype.items():
189189
num_channels = channel_number_dict[stream_name]
190190
file_path = raw_file_paths_dict[stream_name]
191191
size_in_bytes = file_path.stat().st_size
192-
dtype_size = np.dtype(stream_datatype).itemsize
192+
dtype_size = np.dtype(stream_dtype).itemsize
193193
n_samples = size_in_bytes // (dtype_size * num_channels)
194194
signal_stream_memmap = np.memmap(
195195
file_path,
196-
dtype=stream_datatype,
196+
dtype=stream_dtype,
197197
mode="r",
198198
shape=(n_samples, num_channels),
199199
order="C",
@@ -203,9 +203,9 @@ def _parse_header(self):
203203
# for one-file-per-channel we have one memory map / channel stored as a list / neo stream
204204
elif self.file_format == "one-file-per-channel":
205205
self._raw_data = {}
206-
for stream_index, (stream_name, stream_datatype) in enumerate(memmap_data_dtype.items()):
206+
for stream_name, stream_dtype in memmap_data_dtype.items():
207207
channel_file_paths = raw_file_paths_dict[stream_name]
208-
channel_memmap_list = [np.memmap(fp, dtype=stream_datatype, mode="r") for fp in channel_file_paths]
208+
channel_memmap_list = [np.memmap(fp, dtype=stream_dtype, mode="r") for fp in channel_file_paths]
209209
self._raw_data[stream_name] = channel_memmap_list
210210

211211

@@ -216,6 +216,7 @@ def _parse_header(self):
216216

217217
# signals
218218
signal_channels = []
219+
# This is used to calculate bit masks
219220
self.native_channel_order = {}
220221
for chan_info in self._ordered_channel_info:
221222
name = chan_info["custom_channel_name"]

0 commit comments

Comments
 (0)