From 31d836667729b010e77d42da8fc5cac3083a0608 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Fri, 9 Apr 2021 15:42:57 +0200
Subject: [PATCH] running version for DataHandlerWRF and DataHandlerMainSectWRF

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

diff --git a/mlair/data_handler/data_handler_wrf_chem.py b/mlair/data_handler/data_handler_wrf_chem.py
index 1da86143..50bedd15 100644
--- a/mlair/data_handler/data_handler_wrf_chem.py
+++ b/mlair/data_handler/data_handler_wrf_chem.py
@@ -506,7 +506,10 @@ class DataHandlerSingleGridColumn(DataHandlerSingleStation):
 
     @TimeTrackingWrapper
     def extract_data_from_loader(self, loader):
-        data = loader.data.isel(loader.get_nearest_icoords()).squeeze()[self.variables]
+        data = loader.data.isel(loader.get_nearest_icoords()).squeeze(dim=[loader.logical_y_coord_name,
+                                                                           loader.logical_x_coord_name])[self.variables]
+        with ProgressBar():
+            data = dask.compute(data)[0]
         return data
 
     @property
-- 
GitLab