diff --git a/src/probeinterface/neuropixels_tools.py b/src/probeinterface/neuropixels_tools.py index 3e9e513..6cb4499 100644 --- a/src/probeinterface/neuropixels_tools.py +++ b/src/probeinterface/neuropixels_tools.py @@ -865,6 +865,7 @@ def read_spikeglx(file: str | Path) -> Probe: probe.annotate(part_number=imDatPrb_pn) probe.annotate(port=imDatPrb_port) probe.annotate(slot=imDatPrb_slot) + probe.annotate(probe_type=probe_part_number_to_probe_type.get(imDatPrb_pn)) # ===== 8. Set device channel indices (wiring) ===== probe.set_device_channel_indices(np.arange(probe.get_contact_count())) diff --git a/tests/test_io/test_spikeglx.py b/tests/test_io/test_spikeglx.py index 6dce501..570544f 100644 --- a/tests/test_io/test_spikeglx.py +++ b/tests/test_io/test_spikeglx.py @@ -46,6 +46,7 @@ def test_NP1(): assert "1.0" in probe.description assert "adc_group" in probe.contact_annotations assert "adc_sample_order" in probe.contact_annotations + assert probe.annotations.get("probe_type") == "0" def test_NP_phase3A():