diff --git a/nwp/jobs.py b/nwp/jobs.py index 8022b31..c325bbc 100644 --- a/nwp/jobs.py +++ b/nwp/jobs.py @@ -5,21 +5,18 @@ import datetime as dt - - dwd_base_path = "/mnt/storage_b/data/ocf/solar_pv_nowcasting/nowcasting_dataset_pipeline/NWP/DWD" def build_config_on_runtime(model, run, delay=0): config = IconConfig(model=model, run=run, delay=delay, - folder=f"{base_path}/ICON_Global/{run}", - zarr_path=f"{base_path}/ICON_Global/{run}/{run}.zarr.zip"), - config_dict = {"delay": config.delay, "folder": config.folder, "model": config.model, "run": config.run, + folder=f"{base_path}/{'ICON_Global' if model == 'global' else 'ICON_EU'}/{run}", + zarr_path=f"{base_path}/{'ICON_Global' if model == 'global' else 'ICON_EU'}/{run}/{run}.zarr.zip") + config_dict = {"delay": config.delay, "folder": config.folder, "model": config.model, "run": config.run, "zarr_path": config.zarr_path} return config_dict - schedules = [] for r in ["00", "06", "12", "18"]: for model in ["global", "eu"]: