From a5d6489d93e58d2cde2912b0785b0fd56b8649b8 Mon Sep 17 00:00:00 2001 From: Carlos Osuna Date: Fri, 29 May 2026 11:29:45 +0200 Subject: [PATCH 1/3] fix configs and uv requirements --- config/forecasters-ich1.yaml | 2 ++ config/interpolators-ich1.yaml | 2 ++ workflow/rules/inference.smk | 2 +- 3 files changed, 5 insertions(+), 1 deletion(-) diff --git a/config/forecasters-ich1.yaml b/config/forecasters-ich1.yaml index c2b2ee2c..ec515a34 100644 --- a/config/forecasters-ich1.yaml +++ b/config/forecasters-ich1.yaml @@ -17,6 +17,7 @@ runs: - earthkit-utils<0.2.0 - earthkit-data<0.19.0 - git+https://github.com/ecmwf/anemoi-inference.git@main + - anemoi-transform<0.2.0 # forward_transform_filter incompatible with anemoi-transform>=0.2.0 - forecaster: checkpoint: /scratch/mch/apennino/output/checkpoint/973b78b8b39543949abc2b154c5f98d9/inference-last.ckpt @@ -27,6 +28,7 @@ runs: - earthkit-utils<0.2.0 - earthkit-data<0.19.0 - git+https://github.com/ecmwf/anemoi-inference.git@main + - anemoi-transform<0.2.0 # forward_transform_filter incompatible with anemoi-transform>=0.2.0 # - forecaster: # checkpoint: https://huggingface.co/met-no/bris_cloudy-skies/blob/main/2025-10/cloudy-skies-2025-10.ckpt diff --git a/config/interpolators-ich1.yaml b/config/interpolators-ich1.yaml index 24a1fde2..bcce3e48 100644 --- a/config/interpolators-ich1.yaml +++ b/config/interpolators-ich1.yaml @@ -23,10 +23,12 @@ runs: steps: 0/120/6 extra_requirements: - git+https://github.com/ecmwf/anemoi-inference.git@e369b1a90313e9701db13f63364a467aa281cf36 + - anemoi-transform<0.2.0 # e369b1a9 incompatible with anemoi-transform>=0.2.0 - eccodes==2.39.1 - eccodes-cosmo-resources-python==2.38.3.1 extra_requirements: - git+https://github.com/ecmwf/anemoi-inference.git@e369b1a90313e9701db13f63364a467aa281cf36 + - anemoi-transform<0.2.0 # e369b1a9 incompatible with anemoi-transform>=0.2.0 # pinned anemoi-datasets because of ecmwf/anemoi-utils#284, can be removed when fixed - anemoi-datasets==0.5.35 - eccodes==2.39.1 diff --git a/workflow/rules/inference.smk b/workflow/rules/inference.smk index d2b8ef1e..a915ff00 100644 --- a/workflow/rules/inference.smk +++ b/workflow/rules/inference.smk @@ -96,7 +96,7 @@ rule inference_create_venv: source {output.venv}/bin/activate echo "[$(date)] Installing requirements from {input.requirements}..." - uv pip install -r {input.requirements} + uv pip install --index-strategy unsafe-best-match -r {input.requirements} echo "[$(date)] Compiling Python bytecode..." python -m compileall -j 8 -o 0 -o 1 -o 2 .venv/lib/python*/site-packages From 7f2942097516b33685fc837745e67afa94e7d56c Mon Sep 17 00:00:00 2001 From: Carlos Osuna Date: Fri, 29 May 2026 11:30:29 +0200 Subject: [PATCH 2/3] propagate fix to other configs --- config/forecasters-ich1-oper-fixed.yaml | 1 + config/forecasters-ich1-oper.yaml | 1 + 2 files changed, 2 insertions(+) diff --git a/config/forecasters-ich1-oper-fixed.yaml b/config/forecasters-ich1-oper-fixed.yaml index 07b7a102..9284eb25 100644 --- a/config/forecasters-ich1-oper-fixed.yaml +++ b/config/forecasters-ich1-oper-fixed.yaml @@ -19,6 +19,7 @@ runs: config: resources/inference/configs/sgm-multidataset-forecaster-global-ich1-oper.yaml extra_requirements: - git+https://github.com/ecmwf/anemoi-inference.git@b9aaee5df86614cad9d8d08b76876a4be4e980db + - anemoi-transform<0.2.0 # forward_transform_filter incompatible with anemoi-transform>=0.2.0 # - forecaster: # checkpoint: https://service.meteoswiss.ch/mlstore#/experiments/602/runs/233b1274e79b4bfea98378d196a8aba0 diff --git a/config/forecasters-ich1-oper.yaml b/config/forecasters-ich1-oper.yaml index c5b0f76d..13a865b5 100644 --- a/config/forecasters-ich1-oper.yaml +++ b/config/forecasters-ich1-oper.yaml @@ -21,6 +21,7 @@ runs: - earthkit-utils<0.2.0 - earthkit-data<0.19.0 - git+https://github.com/ecmwf/anemoi-inference.git@main + - anemoi-transform<0.2.0 # forward_transform_filter incompatible with anemoi-transform>=0.2.0 - baseline: label: ICON-CH1-ctrl From 07c1995bf061dc4cb1b8368b6c38458d558ee461 Mon Sep 17 00:00:00 2001 From: Carlos Osuna Date: Fri, 29 May 2026 17:57:35 +0200 Subject: [PATCH 3/3] remove the pinned versions of anemoi-transform --- config/forecasters-ich1-oper-fixed.yaml | 1 - config/forecasters-ich1-oper.yaml | 1 - config/forecasters-ich1.yaml | 2 -- config/interpolators-ich1.yaml | 2 -- 4 files changed, 6 deletions(-) diff --git a/config/forecasters-ich1-oper-fixed.yaml b/config/forecasters-ich1-oper-fixed.yaml index 9284eb25..07b7a102 100644 --- a/config/forecasters-ich1-oper-fixed.yaml +++ b/config/forecasters-ich1-oper-fixed.yaml @@ -19,7 +19,6 @@ runs: config: resources/inference/configs/sgm-multidataset-forecaster-global-ich1-oper.yaml extra_requirements: - git+https://github.com/ecmwf/anemoi-inference.git@b9aaee5df86614cad9d8d08b76876a4be4e980db - - anemoi-transform<0.2.0 # forward_transform_filter incompatible with anemoi-transform>=0.2.0 # - forecaster: # checkpoint: https://service.meteoswiss.ch/mlstore#/experiments/602/runs/233b1274e79b4bfea98378d196a8aba0 diff --git a/config/forecasters-ich1-oper.yaml b/config/forecasters-ich1-oper.yaml index 13a865b5..c5b0f76d 100644 --- a/config/forecasters-ich1-oper.yaml +++ b/config/forecasters-ich1-oper.yaml @@ -21,7 +21,6 @@ runs: - earthkit-utils<0.2.0 - earthkit-data<0.19.0 - git+https://github.com/ecmwf/anemoi-inference.git@main - - anemoi-transform<0.2.0 # forward_transform_filter incompatible with anemoi-transform>=0.2.0 - baseline: label: ICON-CH1-ctrl diff --git a/config/forecasters-ich1.yaml b/config/forecasters-ich1.yaml index ec515a34..c2b2ee2c 100644 --- a/config/forecasters-ich1.yaml +++ b/config/forecasters-ich1.yaml @@ -17,7 +17,6 @@ runs: - earthkit-utils<0.2.0 - earthkit-data<0.19.0 - git+https://github.com/ecmwf/anemoi-inference.git@main - - anemoi-transform<0.2.0 # forward_transform_filter incompatible with anemoi-transform>=0.2.0 - forecaster: checkpoint: /scratch/mch/apennino/output/checkpoint/973b78b8b39543949abc2b154c5f98d9/inference-last.ckpt @@ -28,7 +27,6 @@ runs: - earthkit-utils<0.2.0 - earthkit-data<0.19.0 - git+https://github.com/ecmwf/anemoi-inference.git@main - - anemoi-transform<0.2.0 # forward_transform_filter incompatible with anemoi-transform>=0.2.0 # - forecaster: # checkpoint: https://huggingface.co/met-no/bris_cloudy-skies/blob/main/2025-10/cloudy-skies-2025-10.ckpt diff --git a/config/interpolators-ich1.yaml b/config/interpolators-ich1.yaml index bcce3e48..24a1fde2 100644 --- a/config/interpolators-ich1.yaml +++ b/config/interpolators-ich1.yaml @@ -23,12 +23,10 @@ runs: steps: 0/120/6 extra_requirements: - git+https://github.com/ecmwf/anemoi-inference.git@e369b1a90313e9701db13f63364a467aa281cf36 - - anemoi-transform<0.2.0 # e369b1a9 incompatible with anemoi-transform>=0.2.0 - eccodes==2.39.1 - eccodes-cosmo-resources-python==2.38.3.1 extra_requirements: - git+https://github.com/ecmwf/anemoi-inference.git@e369b1a90313e9701db13f63364a467aa281cf36 - - anemoi-transform<0.2.0 # e369b1a9 incompatible with anemoi-transform>=0.2.0 # pinned anemoi-datasets because of ecmwf/anemoi-utils#284, can be removed when fixed - anemoi-datasets==0.5.35 - eccodes==2.39.1