Skip to content
Snippets Groups Projects
Commit 2bfb10cf authored by Felix Kleinert's avatar Felix Kleinert
Browse files

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
parents 30f8a2c6 cee243ea
Branches
Tags
1 merge request!259Draft: Resolve "WRF-Datahandler should inherit from SingleStationDatahandler"
......@@ -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',
)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment