diff --git a/mlair/data_handler/data_handler_wrf_chem.py b/mlair/data_handler/data_handler_wrf_chem.py index 1da8614338f8f6497a344b8194c08fe4e44080de..50bedd15a438dc9814085f9b5c5c70fb31a71bad 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