diff --git a/mlair/data_handler/data_handler_wrf_chem.py b/mlair/data_handler/data_handler_wrf_chem.py index a1b7f96966cb15960096941b408fbd886df45749..092b9ff67144166253bfa5ae8ab2802835ec8ecd 100644 --- a/mlair/data_handler/data_handler_wrf_chem.py +++ b/mlair/data_handler/data_handler_wrf_chem.py @@ -404,7 +404,8 @@ class DataHandlerSingleGridColumn(DataHandlerSingleStation): def __init__(self, *args, external_coords_file=None, var_logical_z_coord_selector=None, - targetvar_logical_z_coord_selector=None, **kwargs): + targetvar_logical_z_coord_selector=None, + rechunk_values=None, **kwargs): self.external_coords_file = external_coords_file self.var_logical_z_coord_selector = self._ret_z_coord_select_if_valid(var_logical_z_coord_selector, as_input=True) @@ -414,6 +415,7 @@ class DataHandlerSingleGridColumn(DataHandlerSingleStation): self._joint_z_coord_selector = self._extract_largest_coord_extractor(self.var_logical_z_coord_selector, self.targetvar_logical_z_coord_selector) self.__loader = None + self.rechunk_values = rechunk_values super().__init__(*args, **kwargs) @staticmethod @@ -463,7 +465,8 @@ class DataHandlerSingleGridColumn(DataHandlerSingleStation): loader = SingleGridColumnWrfChemDataLoader((lat, lon), data_path=path, external_coords_file=self.external_coords_file, - time_dim_name=self.time_dim + time_dim_name=self.time_dim, + rechunk_values=self.rechunk_values ) self.__loader = loader @@ -904,7 +907,7 @@ if __name__ == '__main__': logical_x_coord_name='west_east', logical_y_coord_name='south_north', logical_z_coord_name='bottom_top', - rechunk_values={'XTIME': 1, 'bottom_top': 2}, + rechunk_values={'XTIME': 24, 'bottom_top': 10}, external_coords_file='/home/felix/Data/WRF-Chem/upload_aura_2021-02-24/coords.nc', )