Merge branch...

Merge branch 'felix_issue287_tech-wrf-datahandler-should-inherit-from-singlestationdatahandler' of ssh://gitlab.version.fz-juelich.de:10022/esde/machine-learning/mlair into felix_issue287_tech-wrf-datahandler-should-inherit-from-singlestationdatahandler

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',
                                                     )