handle missing sample_rate from pykilosort params.py

Author: Thinh Nguyen

element_array_ephys/ephys_no_curation.py

@@ -679,7 +679,10 @@ def make(self, key):
         kilosort_dir = find_full_path(get_ephys_root_data_dir(), output_dir)
 
         kilosort_dataset = kilosort.Kilosort(kilosort_dir)
-        acq_software = (EphysRecording & key).fetch1('acq_software')
+        acq_software, sample_rate = (EphysRecording & key).fetch1(
+            'acq_software', 'sampling_rate')
+
+        sample_rate = kilosort_dataset.data['params'].get('sample_rate', sample_rate)
 
         # ---------- Unit ----------
         # -- Remove 0-spike units
@@ -709,7 +712,7 @@ def make(self, key):
             if (kilosort_dataset.data['spike_clusters'] == unit).any():
                 unit_channel, _ = kilosort_dataset.get_best_channel(unit)
                 unit_spike_times = (spike_times[kilosort_dataset.data['spike_clusters'] == unit]
-                                    / kilosort_dataset.data['params']['sample_rate'])
+                                    / sample_rate)
                 spike_count = len(unit_spike_times)
 
                 units.append({