From 13726b29dae0145308906258c7e703d712622fc9 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Wed, 5 Jan 2022 14:46:08 +0100
Subject: [PATCH] squeeze upstream wind sector

---
 mlair/run_modules/post_processing.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py
index 5691624f..297f378d 100644
--- a/mlair/run_modules/post_processing.py
+++ b/mlair/run_modules/post_processing.py
@@ -158,9 +158,10 @@ class PostProcessing(RunEnvironment):
             ds = ds.rename({'XTIME': self.index_dim})
         except ValueError as e:
             logging.warning("Dimension `XTIME' does not exist")
+        ds = ds.squeeze()
         logging.info(f"PostProcessing.load_upstream_wind_sector({name_of_set}: shape of `upstream_wind_sector' is "
                      f"{ds.shape}\ndims are {ds.dims}")
-        self.upstream_wind_sector = ds.squeeze()
+        self.upstream_wind_sector = ds
 
     @TimeTrackingWrapper
     def estimate_sample_uncertainty(self, separate_ahead=False):
-- 
GitLab