From 8782f7284a3588fcd3d2b79b733d8d67aa3aeeb7 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Wed, 5 Jan 2022 15:22:25 +0100
Subject: [PATCH] squeeze coords of upstream wind sectors in wrfdh
---
mlair/data_handler/data_handler_wrf_chem.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/mlair/data_handler/data_handler_wrf_chem.py b/mlair/data_handler/data_handler_wrf_chem.py
index 15837977..2d0ea525 100644
--- a/mlair/data_handler/data_handler_wrf_chem.py
+++ b/mlair/data_handler/data_handler_wrf_chem.py
@@ -1253,7 +1253,8 @@ class DataHandlerSectorGrid(DataHandlerSingleGridColumn):
f"data/{self.station[0]}_{self.start}_{self.end}_upstream_wind_sector.nc")
wind_upstream_sector_by_name = self.wind_upstream_sector_by_name
dims_to_expand = list(wind_upstream_sector_by_name.coords._names - set(wind_upstream_sector_by_name.dims))
- wind_upstream_sector_by_name = wind_upstream_sector_by_name.expand_dims(dims_to_expand)
+ wind_upstream_sector_by_name = wind_upstream_sector_by_name.expand_dims(dims_to_expand).squeeze(
+ dim=list(set(dims_to_expand) - {self.iter_dim, self.target_dim}))
wind_upstream_sector_by_name = wind_upstream_sector_by_name.to_dataset(self.iter_dim)
wind_upstream_sector_by_name.to_netcdf(file_name)
--
GitLab