diff --git a/src/run_modules/pre_processing.py b/src/run_modules/pre_processing.py index a32f7075aaca39f59fffb0a2547a4dcd0cc5e6d5..af66c772bb6d4c35505f1f08ce688135af57f3e8 100644 --- a/src/run_modules/pre_processing.py +++ b/src/run_modules/pre_processing.py @@ -13,7 +13,7 @@ from src.join import EmptyQueryResult DEFAULT_ARGS_LIST = ["data_path", "network", "stations", "variables", "interpolate_dim", "target_dim", "target_var"] DEFAULT_KWARGS_LIST = ["limit_nan_fill", "window_history_size", "window_lead_time", "statistics_per_var", - "station_type", "overwrite_local_data"] + "station_type", "overwrite_local_data", "start", "end"] class PreProcessing(RunEnvironment): @@ -39,6 +39,7 @@ class PreProcessing(RunEnvironment): valid_stations = self.check_valid_stations(args, kwargs, self.data_store.get("stations", "general")) self.data_store.put("stations", valid_stations, "general") self.split_train_val_test() + self.report_pre_processing() def report_pre_processing(self): logging.info(20 * '##')