Skip to content

Commit cad48e4

Browse files
Fixed bug for .prb files with np statements
1 parent f90c617 commit cad48e4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

kilosort/io.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,10 @@ def load_probe(probe_path):
6363
# !DOES NOT WORK FOR PHASE3A PROBES WITH DISCONNECTED CHANNELS!
6464
# Also does not remove reference channel in PHASE3B probes
6565
contents = probe_path.read_text()
66-
metadata = {}
66+
metadata = {'np': np}
6767
exec(contents, {}, metadata)
68+
# TODO: Figure out an alternative for exec, it's a security risk.
69+
6870
probe['chanMap'] = []
6971
probe['xc'] = []
7072
probe['yc'] = []

0 commit comments

Comments
 (0)