Fix pca transform error

SpikeInterface · Jul 10, 2024 · caefa4e · caefa4e
1 parent e4fa25a
commit caefa4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/spikeinterface/postprocessing/principal_component.py b/src/spikeinterface/postprocessing/principal_component.py
@@ -493,7 +493,7 @@ def _transform_waveforms(self, spikes, waveforms, pca_model, progress_bar):
                     try:
                         proj = pca_model.transform(wfs[:, :, wf_ind])
                         pca_projection[:, :, wf_ind][spike_mask, :] = proj
-                    except NotFittedError as e:
+                    except:
                         # this could happen if len(wfs) is less then n_comp for a channel
                         project_on_non_fitted = True
             if project_on_non_fitted: