From 483bc0b3b5234e71db17ffbafcf349877f004ba3 Mon Sep 17 00:00:00 2001 From: Peter Dudfield Date: Mon, 30 Mar 2026 21:00:42 +0100 Subject: [PATCH] fix if model doesnt have prob --- src/dataplatform/forecast/data.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/dataplatform/forecast/data.py b/src/dataplatform/forecast/data.py index 4799044..947edf2 100644 --- a/src/dataplatform/forecast/data.py +++ b/src/dataplatform/forecast/data.py @@ -116,9 +116,10 @@ async def get_forecast_data_one_forecaster( # get plevels into columns and rename them 'fraction columns_before_expand = set(all_data_df.columns) - all_data_df = all_data_df.pipe( - lambda df: df.join(pd.json_normalize(df["other_statistics_fractions"])), - ).drop("other_statistics_fractions", axis=1) + if "other_statistics_fractions" in all_data_df.columns: + all_data_df = all_data_df.pipe( + lambda df: df.join(pd.json_normalize(df["other_statistics_fractions"])), + ).drop("other_statistics_fractions", axis=1) new_columns = set(all_data_df.columns) - columns_before_expand if len(new_columns) > 0: all_data_df = all_data_df.rename(columns={col: f"{col}_fraction" for col in new_columns})