diff --git a/video_prediction_tools/postprocess/statistical_evaluation.py b/video_prediction_tools/postprocess/statistical_evaluation.py index e011b4016b134ea3986432c67d7b9b6c720e9432..6e5da00dd06a1fc10c953dd76fb5053de66a4e34 100644 --- a/video_prediction_tools/postprocess/statistical_evaluation.py +++ b/video_prediction_tools/postprocess/statistical_evaluation.py @@ -76,7 +76,7 @@ def calculate_cond_quantiles(data_fcst: xr.DataArray, data_ref: xr.DataArray, fa bins_all = list(np.arange(int(data_all_min), int(data_all_max) + 1)) bins_c_all = 0.5 * (np.asarray(bins_all[0:-1]) + np.asarray(bins_all[1:])) # initialize quantile data array - quantile_panel = xr.DataArray(np.full((nbins_all, nquantiles), np.nan), + quantile_panel = xr.DataArray(np.full((len(bins_c_all), nquantiles), np.nan), coords={"bin_center": bins_c_all, "quantile": list(quantiles)}, dims=["bin_center", "quantile"], attrs={"cond_var_name": data_cond_longname, "cond_var_unit": data_cond_unit,