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