From 168cca2f7f3442620041596d9468047485c4d800 Mon Sep 17 00:00:00 2001
From: RoyStegeman <roystegeman@live.nl>
Date: Fri, 18 Aug 2023 11:05:58 +0100
Subject: [PATCH] don't print msr layer summary if msr not enforced

.
---
 n3fit/src/n3fit/model_trainer.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/n3fit/src/n3fit/model_trainer.py b/n3fit/src/n3fit/model_trainer.py
index 2f77590a6e..b96737481c 100644
--- 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,