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(