Skip to content

Commit 3e8297d

Browse files
committed
Fix kcoords and tests
1 parent 5c6b54f commit 3e8297d

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

.github/scripts/test_kilosort4_ci.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@
114114

115115
if parse(kilosort.__version__) >= parse("4.0.37"):
116116
PARAMS_TO_TEST_DICT.update({"max_cluster_subset": 20})
117+
PARAMETERS_NOT_AFFECTING_RESULTS.append("max_cluster_subset")
117118

118119

119120
PARAMS_TO_TEST = list(PARAMS_TO_TEST_DICT.keys())

src/spikeinterface/sorters/external/kilosort4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -440,7 +440,7 @@ def _setup_json_probe_map(cls, recording, sorter_output_folder):
440440
yc = positions[:, 1]
441441
unique_groups = set(groups)
442442
group_map = {group: idx for idx, group in enumerate(unique_groups)}
443-
kcoords = np.array(group_map.values(), dtype=int)
443+
kcoords = np.array(list(group_map.values()), dtype=int)
444444

445445
probe = {
446446
"chanMap": chanMap,

0 commit comments

Comments
 (0)