clean up dry_run print

LSSTDESC · Aug 20, 2024 · 70e40a1 · 70e40a1
1 parent 1d5c17e
commit 70e40a1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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(