Merge pull request #2102 from alejoe91/fix-openephys.multi-shank

Fix grouping of OpenEphys NPIX
SpikeInterface · Oct 16, 2023 · 05419fc · 05419fc
2 parents 6b04e29 + 9addc57
commit 05419fc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/modules_gallery/qualitymetrics/plot_4_curation.py b/examples/modules_gallery/qualitymetrics/plot_4_curation.py
@@ -61,4 +61,4 @@
 
 curated_sorting = sorting.select_units(keep_unit_ids)
 print(curated_sorting)
-se.NpzSortingExtractor.write_sorting(curated_sorting, 'curated_sorting.pnz')
+se.NpzSortingExtractor.write_sorting(curated_sorting, 'curated_sorting.npz')
diff --git a/src/spikeinterface/extractors/neoextractors/openephys.py b/src/spikeinterface/extractors/neoextractors/openephys.py
@@ -183,7 +183,10 @@ def __init__(
                 probe = None
 
             if probe is not None:
-                self = self.set_probe(probe, in_place=True)
+                if probe.shank_ids is not None:
+                    self.set_probe(probe, in_place=True, group_mode="by_shank")
+                else:
+                    self.set_probe(probe, in_place=True)
                 probe_name = probe.annotations["probe_name"]
                 # load num_channels_per_adc depending on probe type
                 if "2.0" in probe_name: