Add station obs from peakweather

Author: dnerini

pyproject.toml

@@ -24,6 +24,7 @@ dependencies = [
     "earthkit-plots",
     "marimo>=0.16.5",
     "geopandas>=0.14.0",
+    "peakweather",
 ]
 
 [project.optional-dependencies]
@@ -58,3 +59,6 @@ packages = [
   "src/verification",
   "src/data_input"
 ]
+
+[tool.uv.sources]
+peakweather = { git = "https://github.com/MeteoSwiss/PeakWeather.git" }

uv.lock

@@ -56,3 +56,14 @@ if "extract_cosmo1e" in config.get("include-optional-rules", []):
                 --steps {params.steps} \
                     > {log} 2>&1
             """
+
+
+rule download_obs_from_peakweather:
+    localrule: True
+    output:
+        peakweather=directory(OUT_ROOT / "data/observations/peakweather"),
+    run:
+        from peakweather.dataset import PeakWeatherDataset
+
+        # Download the data from Huggingface
+        ds = PeakWeatherDataset(root=output.peakweather)

workflow/rules/data.smk

@@ -26,6 +26,7 @@ rule plot_meteogram:
         inference_okfile=rules.execute_inference.output.okfile,
         analysis_zarr=config["analysis"].get("analysis_zarr"),
         baseline_zarr=lambda wc: _use_first_baseline_zarr(wc),
+        peakweather_dir=rules.download_obs_from_peakweather.output.peakweather,
     output:
         OUT_ROOT / "showcases/{run_id}/{init_time}/{init_time}_{param}_{sta}.png",
     # localrule: True
@@ -42,13 +43,13 @@ rule plot_meteogram:
         export ECCODES_DEFINITION_PATH=$(realpath .venv/share/eccodes-cosmo-resources/definitions)
         python {input.script} \
             --forecast {params.grib_out_dir}  --analysis {input.analysis_zarr} \
-            --baseline {input.baseline_zarr} \
+            --baseline {input.baseline_zarr} --peakweather {input.peakweather_dir} \
             --date {wildcards.init_time} --outfn {output[0]} \
             --param {wildcards.param}  --station {wildcards.sta}
         # interactive editing (needs to set localrule: True and use only one core)
         # marimo edit {input.script} -- \
         #     --forecast {params.grib_out_dir}  --analysis {input.analysis_zarr} \
-        #     --baseline {input.baseline_zarr} \
+        #     --baseline {input.baseline_zarr} --peakweather {input.peakweather_dir} \
         #     --date {wildcards.init_time} --outfn {output[0]} \
         #     --param {wildcards.param}  --station {wildcards.sta}
         """