From 179c334b6e1737e7836c3997a994dc21c6f3609e Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Fri, 19 Apr 2024 14:38:57 +0100 Subject: [PATCH] add switch to turn off scenario data (for carbon and heat for now) --- .../ml-pipeline/src/pipeline/5_generate_scenarios.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/modules/ml-pipeline/src/pipeline/5_generate_scenarios.py b/modules/ml-pipeline/src/pipeline/5_generate_scenarios.py index 9d2fa68..6debe32 100644 --- a/modules/ml-pipeline/src/pipeline/5_generate_scenarios.py +++ b/modules/ml-pipeline/src/pipeline/5_generate_scenarios.py @@ -78,6 +78,18 @@ def generate_scenario_predictions( scenario_data = pd.DataFrame() + # If we have no scenario data, we can save empty dataframes + if scenario_data_filepaths is None: + logger.info("No scenario data filepaths provided") + output_dataclient.save_data( + obj=scenario_data, location=comparison_output_filepath, save_config=None + ) + + output_dataclient.save_data( + obj=scenario_data, location=metrics_output_filepath, save_config=None + ) + return + # Can have multiple scenario data files for scenario_data_filepath in scenario_data_filepaths: scenario_data = pd.concat(