supporting neuropixels UHD in ephys.EphysRecording

Author: Thinh Nguyen

element_array_ephys/ephys_no_curation.py

@@ -240,6 +240,8 @@ def make(self, key):
                 f'Ephys recording data not found!'
                 f' Neither SpikeGLX nor Open Ephys recording files found')
 
+        supported_probe_types = probe.ProbeType.fetch('probe_type')
+
         if acq_software == 'SpikeGLX':
             for meta_filepath in ephys_meta_filepaths:
                 spikeglx_meta = spikeglx.SpikeGLXMeta(meta_filepath)
@@ -249,7 +251,7 @@ def make(self, key):
                 raise FileNotFoundError(
                     'No SpikeGLX data found for probe insertion: {}'.format(key))
 
-            if re.search('(1\.0|2\.0)', spikeglx_meta.probe_model):
+            if spikeglx_meta.probe_model in supported_probe_types:
                 probe_type = spikeglx_meta.probe_model
                 electrode_query = probe.ProbeType.Electrode & {'probe_type': probe_type}
 
@@ -288,7 +290,7 @@ def make(self, key):
                 raise FileNotFoundError(
                     'No Open Ephys data found for probe insertion: {}'.format(key))
 
-            if re.search('(1|2)\.0', probe_data.probe_model):
+            if probe_data.probe_model in supported_probe_types:
                 probe_type = probe_data.probe_model
                 electrode_query = probe.ProbeType.Electrode & {'probe_type': probe_type}