Fix for issue #56

3dem · Nov 22, 2023 · d258f2f · d258f2f
1 parent e3cf6a6
commit d258f2f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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}"