Describe the bug
When calling init for ERA5Forecasting, the arguments passed to it's parent class are (root_dir, root_highres_dir, in_vars, years, split). Because of this while creating the data_dict , only variables that are part of in_vars are loaded from netcdf files. See lines 45 and 61.
This can result in potential bug at line 122 when we create the output data, if the output variables are not a subset of input variables.
Describe the bug When calling init for
ERA5Forecasting, the arguments passed to it's parent class are(root_dir, root_highres_dir, in_vars, years, split). Because of this while creating the data_dict , only variables that are part of in_vars are loaded from netcdf files. See lines 45 and 61.This can result in potential bug at line 122 when we create the output data, if the output variables are not a subset of input variables.