From d258f2f546a61a3ff239cba48bb4a6bc76ee9512 Mon Sep 17 00:00:00 2001
From: Kiarash Jamali <kjamali@mrc-lmb.cam.ac.uk>
Date: Wed, 22 Nov 2023 05:23:50 +0000
Subject: [PATCH] Fix for issue #56

---
 model_angelo/apps/build.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/model_angelo/apps/build.py b/model_angelo/apps/build.py
index 26bd042..3ea33cd 100644
--- a/model_angelo/apps/build.py
+++ b/model_angelo/apps/build.py
@@ -234,7 +234,8 @@ def main(parsed_args):
             gnn_infer_args.write_hmm_profiles = False
             gnn_infer_args.refine = False
 
-            gnn_infer_args.aggressive_pruning = True
+            if parsed_args.config_path is None:
+                gnn_infer_args.aggressive_pruning = True
 
             logger.info(
                 f"GNN model refinement round {i + 1} with args: {gnn_infer_args}"