From 70e40a140fcdf64e0794e549c4b1415539e4cd4d Mon Sep 17 00:00:00 2001
From: Eric Charles <badass@stanford.edu>
Date: Tue, 20 Aug 2024 14:59:40 -0700
Subject: [PATCH] clean up dry_run print

---
 src/rail/cli/commands.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/rail/cli/commands.py b/src/rail/cli/commands.py
index 30f7be5..f4c2744 100644
--- a/src/rail/cli/commands.py
+++ b/src/rail/cli/commands.py
@@ -124,7 +124,10 @@ def estimate(stage_name, stage_class, stage_module, stages_config, model_file, c
     if dry_run:
         print(f"Running stage {stage.name} of type {type(stage)} on {input_file}")
         print("Stage config:   ")
-        pprint.pprint(stage.config.to_dict())
+        print_dict = stage.config.to_dict().copy()
+        for rm_key in ['config', 'inputs', 'stages']:
+            print_dict.pop(rm_key, None)
+        pprint.pprint(print_dict)
         return 0
         
     output = PZFactory.run_cat_estimator_stage(