From 1faf395d5510ebf1e8ec5a4998cd3adfb28a9d98 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Fri, 9 Apr 2021 13:22:17 +0200
Subject: [PATCH] include 'stagger' fix as cdo removed empty attrs

---
 mlair/data_handler/data_handler_wrf_chem.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mlair/data_handler/data_handler_wrf_chem.py b/mlair/data_handler/data_handler_wrf_chem.py
index 36fbb0c7..6d5f58f0 100644
--- a/mlair/data_handler/data_handler_wrf_chem.py
+++ b/mlair/data_handler/data_handler_wrf_chem.py
@@ -255,6 +255,12 @@ class BaseWrfChemDataLoader:
             u_target_field = self._data[mapping_of_stag2unstag[u_stag_name]]
             v_target_field = self._data[mapping_of_stag2unstag[v_stag_name]]
 
+            # This if statements are needed as cdo might have removed empty attrs from variables
+            if "stagger" not in u_staged_field.attrs:
+                u_staged_field.attrs['stagger'] = ""
+            if "stagger" not in v_staged_field.attrs:
+                v_staged_field.attrs['stagger'] = ""
+
             u_stagger = u_staged_field.attrs['stagger']
             v_stagger = v_staged_field.attrs['stagger']
 
-- 
GitLab