Still run if there's no post_process_config in config.yaml

switch-model · Jan 28, 2023 · f962b67 · f962b67
1 parent 03d97b4
commit f962b67
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)