From 0c23a584952260dba5ff3edb93b335d55246e618 Mon Sep 17 00:00:00 2001
From: Danilo Ferreira de Lima <danilo.enoque.ferreira.de.lima@xfel.de>
Date: Tue, 31 Jan 2023 18:00:37 +0100
Subject: [PATCH] Corrected compatibility check.

---
 pes_to_spec/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py
index 55bd17d..0ed420a 100644
--- a/pes_to_spec/model.py
+++ b/pes_to_spec/model.py
@@ -652,9 +652,9 @@ class Model(TransformerMixin, BaseEstimator):
         """
         selection_model = self.x_model['select']
         low_res = selection_model.transform(low_res_data, keep_dictionary_structure=True)
-        pca_model = self.channel_pca_model.named_steps['pca']
         quality = {channel: 0.0 for channel in low_res.keys()}
         for channel in low_res.keys():
+            pca_model = self.channel_pca_model[channel].named_steps['pca']
             low_pca = pca_model.transform(low_res[channel])
             low_pca_rec = pca_model.inverse_transform(low_pca)
             low_pca_unc = self.channel_pca_model.named_steps['unc'].uncertainty()
-- 
GitLab