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

add back fix for short probes
int-brain-lab · Dec 8, 2023 · 74e6fd5 · 74e6fd5
2 parents 679d9c5 + e06ba40
commit 74e6fd5
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))