diff --git a/.github/scripts/test_kilosort4_ci.py b/.github/scripts/test_kilosort4_ci.py index df4cb64216..1da2f2ba92 100644 --- a/.github/scripts/test_kilosort4_ci.py +++ b/.github/scripts/test_kilosort4_ci.py @@ -569,6 +569,7 @@ def monkeypatch_filter_function(self, X, ops=None, ibatch=None): results["si"]["st"] = np.load(spikeinterface_output_dir / "sorter_output" / "spike_times.npy") results["si"]["clus"] = np.load(spikeinterface_output_dir / "sorter_output" / "spike_clusters.npy") assert np.allclose(results["ks"]["st"], results["si"]["st"], rtol=0, atol=1) + assert np.array_equal(results["ks"]["clus"], results["si"]["clus"]) ##### Helpers ###### def _get_kilosort_native_settings(self, recording, paths, param_key, param_value):