diff --git a/.github/import_test.py b/.github/import_test.py index 9c26736e18..f7c3e9f858 100644 --- a/.github/import_test.py +++ b/.github/import_test.py @@ -46,7 +46,7 @@ time_taken_list.append(time_taken) for time in time_taken_list: - if time > 2.5: + if time > 1.5: exceptions.append(f"Importing {import_statement} took too long: {time:.2f} seconds") break diff --git a/src/spikeinterface/qualitymetrics/pca_metrics.py b/src/spikeinterface/qualitymetrics/pca_metrics.py index bd23268b3b..2915cee8ec 100644 --- a/src/spikeinterface/qualitymetrics/pca_metrics.py +++ b/src/spikeinterface/qualitymetrics/pca_metrics.py @@ -667,6 +667,8 @@ def nearest_neighbors_noise_overlap( ---------- Based on noise overlap metric described in [Chung]_ """ + from sklearn.decomposition import IncrementalPCA + rng = np.random.default_rng(seed=seed) waveforms_ext = sorting_analyzer.get_extension("waveforms")