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(