Merge pull request #111 from int-brain-lab/develop

Develop
int-brain-lab · Jan 12, 2024 · 3e60f33 · 3e60f33
2 parents b122b77 + 5e92d9a
commit 3e60f33
Show file tree

Hide file tree

Showing 2 changed files with 301 additions and 47 deletions.
diff --git a/atlaselectrophysiology/plot_data.py b/atlaselectrophysiology/plot_data.py
@@ -600,7 +600,12 @@ def get_autocorr(self, clust_idx):
         autocorr = xcorr(self.data['spikes']['times'][idx], self.data['spikes']['clusters'][idx],
                          AUTOCORR_BIN_SIZE, AUTOCORR_WIN_SIZE)
 
-        return autocorr[0, 0, :], self.data['clusters'].metrics.cluster_id[self.clust_id[clust_idx]]
+        if self.data['clusters'].get('metrics', {}).get('cluster_id', None) is None:
+            clust_id = self.clust_id[clust_idx]
+        else:
+            clust_id = self.data['clusters'].metrics.cluster_id[self.clust_id[clust_idx]]
+
+        return autocorr[0, 0, :], clust_id
 
     def get_template_wf(self, clust_idx):
         template_wf = (self.data['clusters']['waveforms'][self.clust_id[clust_idx], :, 0])