Skip to content

Commit d11be08

Browse files
committed
fix neuronexus buffer_id
1 parent b8a60c5 commit d11be08

2 files changed

Lines changed: 10 additions & 1 deletion

File tree

neo/rawio/neuronexusrawio.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
BaseRawIO,
4747
_signal_channel_dtype,
4848
_signal_stream_dtype,
49+
_signal_buffer_dtype,
4950
_spike_channel_dtype,
5051
_event_channel_dtype,
5152
)
@@ -196,11 +197,14 @@ def _parse_header(self):
196197

197198
signal_channels = np.array(signal_channels, dtype=_signal_channel_dtype)
198199

200+
buffer_id = "0"
201+
signal_buffers = np.array([("", buffer_id)], dtype=_signal_buffer_dtype)
202+
199203
stream_ids = np.unique(signal_channels["stream_id"])
200204
signal_streams = np.zeros(stream_ids.size, dtype=_signal_stream_dtype)
201205
signal_streams["id"] = [str(stream_id) for stream_id in stream_ids]
202206
# One unique buffer
203-
signal_streams["buffer_id"] = "0"
207+
signal_streams["buffer_id"] = buffer_id
204208

205209
for stream_index, stream_id in enumerate(stream_ids):
206210
name = stream_id_to_stream_name.get(int(stream_id), "")
@@ -218,6 +222,7 @@ def _parse_header(self):
218222
self.header = {}
219223
self.header["nb_block"] = 1
220224
self.header["nb_segment"] = [1]
225+
self.header["signal_buffers"] = signal_buffers
221226
self.header["signal_streams"] = signal_streams
222227
self.header["signal_channels"] = signal_channels
223228
self.header["spike_channels"] = spike_channels

neo/test/rawiotest/test_neuronexusrawio.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,3 +12,7 @@ class TestNeuroNexusRawIO(
1212
rawioclass = NeuroNexusRawIO
1313
entities_to_download = ["neuronexus"]
1414
entities_to_test = ["neuronexus/allego_1/allego_2__uid0701-13-04-49.xdat.json"]
15+
16+
17+
if __name__ == "__main__":
18+
unittest.main()

0 commit comments

Comments
 (0)