From 41211579d6a3e774118f43ec437b7b594c8c916b Mon Sep 17 00:00:00 2001
From: "v.gramlich1" <v.gramlichfz-juelich.de>
Date: Fri, 23 Jul 2021 10:57:52 +0200
Subject: [PATCH] Enabled window_lead_time=1
---
mlair/plotting/postprocessing_plotting.py | 2 ++
mlair/run_modules/post_processing.py | 9 ++++++++-
2 files changed, 10 insertions(+), 1 deletion(-)
diff --git a/mlair/plotting/postprocessing_plotting.py b/mlair/plotting/postprocessing_plotting.py
index 491aa52e..cd898e9e 100644
--- a/mlair/plotting/postprocessing_plotting.py
+++ b/mlair/plotting/postprocessing_plotting.py
@@ -81,6 +81,8 @@ class PlotMonthlySummary(AbstractPlotClass):
data_nn = data.sel(type=self._model_name).squeeze()
if len(data_nn.shape) > 1:
data_nn = data_nn.assign_coords(ahead=[f"{days}d" for days in data_nn.coords["ahead"].values])
+ else:
+ data_nn.coords["ahead"].values = str(data_nn.coords["ahead"].values) + "d"
data_obs = data.sel(type="obs", ahead=1).squeeze()
data_obs.coords["ahead"] = "obs"
diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py
index 0d7bfeb4..2c31fba9 100644
--- a/mlair/run_modules/post_processing.py
+++ b/mlair/run_modules/post_processing.py
@@ -533,7 +533,14 @@ class PostProcessing(RunEnvironment):
"""
tmp_ols = self.ols_model.predict(input_data)
target_shape = ols_prediction.values.shape
- ols_prediction.values = np.swapaxes(tmp_ols, 2, 0) if target_shape != tmp_ols.shape else tmp_ols
+ if target_shape != tmp_ols.shape:
+ if len(target_shape)==2:
+ new_values = np.swapaxes(tmp_ols,1,0)
+ else:
+ new_values = np.swapaxes(tmp_ols, 2, 0)
+ else:
+ new_values = tmp_ols
+ ols_prediction.values = new_values
if not normalised:
ols_prediction = transformation_func(ols_prediction, "target", inverse=True)
return ols_prediction
--
GitLab