add back fix for short probes

int-brain-lab · Dec 8, 2023 · e06ba40 · e06ba40
1 parent f105fcb
commit e06ba40
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/atlaselectrophysiology/plot_data.py b/atlaselectrophysiology/plot_data.py
@@ -407,6 +407,9 @@ def gain2level(gain):
                 h = neuropixel.split_trace_header(h, shank=int(sr.meta.get('NP2.4_shank')))
             else:
                 h = neuropixel.trace_header(sr.major_version, nshank=np.unique(th['shank']).size)
+                idx = np.isin(h['ind'], th['ind'])
+                for k in h.keys():
+                    h[k] = h[k][idx]
 
             s0 = t * sr.fs
             tsel = slice(int(s0), int(s0) + int(1 * sr.fs))