From 0ed4876dfedb175ca08c90d94c8a1d3e215d0586 Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Wed, 4 Sep 2024 14:25:57 +0200 Subject: [PATCH] Extend check on clus --- .github/scripts/test_kilosort4_ci.py | 1 + 1 file changed, 1 insertion(+) 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):