diff --git a/switch_model/wecc/get_inputs/cli.py b/switch_model/wecc/get_inputs/cli.py index 29db6588c..4ec694aac 100644 --- a/switch_model/wecc/get_inputs/cli.py +++ b/switch_model/wecc/get_inputs/cli.py @@ -65,7 +65,9 @@ def run_post_process(module): post_process = getattr(mod, "post_process") # Get specific configuration for the post process if specified - post_config = full_config["post_process_config"].get(module, None) + post_config = None + if "post_process_config" in full_config and full_config["post_process_config"] is not None: + post_config = full_config["post_process_config"].get(module, None) # Run post process post_process(post_config)