From 5f916240cf5422813bd78894ad72f3650470c1cd Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Thu, 18 Jul 2024 09:05:17 +0200 Subject: [PATCH] PCA: Extend try-except to fit in run_for_all_spikes --- src/spikeinterface/postprocessing/principal_component.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/spikeinterface/postprocessing/principal_component.py b/src/spikeinterface/postprocessing/principal_component.py index 1138adac7d..99c60a5043 100644 --- a/src/spikeinterface/postprocessing/principal_component.py +++ b/src/spikeinterface/postprocessing/principal_component.py @@ -637,7 +637,11 @@ def _all_pc_extractor_chunk(segment_index, start_frame, end_frame, worker_ctx): w = wf[:, chan_ind] if w.size > 0: w = w[None, :] - all_pcs[i, :, c] = pca_model[chan_ind].transform(w) + try: + all_pcs[i, :, c] = pca_model[chan_ind].transform(w) + except: + # this could happen if len(wfs) is less then n_comp for a channel + pass def _init_work_all_pc_extractor(recording, sorting, all_pcs_args, nbefore, nafter, unit_channels, pca_model):