Update path (#5)

openclimatefix · Sep 6, 2023 · 41070db · 41070db
1 parent 8c22682
commit 41070db
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nwp/jobs.py b/nwp/jobs.py
@@ -2,15 +2,15 @@
 
 from nwp.assets.dwd.common import IconConfig
 
-base_path = "/mnt/storage_b/data/ocf/data/ocf/solar_pv_nowcasting/nowcasting_dataset_pipeline/NWP/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")
+                        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