Update preprocessing layer in vpinterface

NNPDF · Dec 5, 2023 · 1cb492c · 1cb492c
1 parent 0f198ef
commit 1cb492c
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/n3fit/src/n3fit/vpinterface.py b/n3fit/src/n3fit/vpinterface.py
@@ -224,13 +224,7 @@ def get_preprocessing_factors(self, replica=None):
         if replica is None:
             replica = 1
         # Replicas start counting in 1 so:
-        preprocessing_layers = self._models[replica - 1].get_layer_re(r"preprocessing_factor_\d")
-        if len(preprocessing_layers) > 1:
-            # We really don't want to fail at this point, but print a warning at least...
-            log.warning("More than one preprocessing layer found within the model!")
-        elif len(preprocessing_layers) < 1:
-            log.warning("No preprocessing layer found within the model!")
-        preprocessing_layer = preprocessing_layers[0]
+        preprocessing_layer = self._models[replica - 1].get_layer("preprocessing_factor")
 
         alphas_and_betas = None
         if self.fit_basis is not None: