don't print msr layer summary if msr not enforced

.
NNPDF · Aug 18, 2023 · 168cca2 · 168cca2
1 parent 9cf5f9e
commit 168cca2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/n3fit/src/n3fit/model_trainer.py b/n3fit/src/n3fit/model_trainer.py
@@ -476,8 +476,12 @@ def _model_generation(self, xinput, pdf_models, partition, partition_idx):
             pdf_model.summary()
             nn_model = pdf_model.get_layer("NN_0")
             nn_model.summary()
-            msr_model = pdf_model.get_layer("impose_msr")
-            msr_model.summary()
+            # We may have fits without sumrules imposed
+            try:
+                msr_model = pdf_model.get_layer("impose_msr")
+                msr_model.summary()
+            except ValueError:
+                pass
 
         models = {
             "training": training,