handles single probe recording in "Neuropix-PXI" format

Author: Thinh Nguyen

element_array_ephys/readers/openephys.py

@@ -211,7 +211,7 @@ def __init__(self, processor, probe_index=0):
             self._channels_connected = {int(re.search(r'\d+$', k).group()): int(v)
                                         for k, v in self.probe_info.pop('CHANNELSTATUS').items()}
         else:  # Neuropix-PXI
-            self.probe_info = processor['EDITOR']['NP_PROBE'][probe_index]
+            self.probe_info = processor['EDITOR']['NP_PROBE'] if isinstance(processor['EDITOR']['NP_PROBE'], dict) else processor['EDITOR']['NP_PROBE'][probe_index]
             self.probe_SN = self.probe_info['@probe_serial_number']
             self.probe_model = _probe_model_name_mapping[self.probe_info['@probe_name']]