Skip to content

Commit 686ed53

Browse files
committed
Fix tests for 4.0.37
1 parent 64cee6d commit 686ed53

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

.github/scripts/test_kilosort4_ci.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,9 @@
112112
PARAMS_TO_TEST_DICT.update({"cluster_neighbors": 11})
113113
PARAMETERS_NOT_AFFECTING_RESULTS.append("cluster_neighbors")
114114

115+
if parse(kilosort.__version__) >= parse("4.0.37"):
116+
PARAMS_TO_TEST_DICT.update({"max_cluster_subset": 20})
117+
115118

116119
PARAMS_TO_TEST = list(PARAMS_TO_TEST_DICT.keys())
117120

@@ -254,6 +257,8 @@ def test_initialize_ops_arguments(self):
254257
"device",
255258
"save_preprocessed_copy",
256259
]
260+
if parse(kilosort.__version__) >= parse("4.0.37"):
261+
expected_arguments += ["gui_mode"]
257262

258263
self._check_arguments(
259264
initialize_ops,

0 commit comments

Comments
 (0)