From 51bf1d45d3a7bcae2c47ea84dd44b1e4582b24b1 Mon Sep 17 00:00:00 2001 From: Francesco Zanetta Date: Wed, 10 Dec 2025 10:44:36 +0100 Subject: [PATCH 1/3] simplify pyproject template --- workflow/envs/anemoi_inference.toml | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/workflow/envs/anemoi_inference.toml b/workflow/envs/anemoi_inference.toml index 982a673..cac570f 100644 --- a/workflow/envs/anemoi_inference.toml +++ b/workflow/envs/anemoi_inference.toml @@ -6,13 +6,12 @@ dependencies = [ "torch", "torchvision", "torchaudio", - "anemoi-datasets>=0.5.23,<0.7.0", - "anemoi-graphs>=0.5.0,<0.7.0", - "anemoi-inference>=0.8.0,<0.9.0", - "anemoi-models>=0.4.20,<0.6.0", - "anemoi-training>=0.3.3,<0.5.0", - "anemoi-transform>=0.1.10,<0.3.0", - "anemoi-utils>=0.4.22,<0.5.0", + "anemoi-datasets", + "anemoi-graphs", + "anemoi-inference", + "anemoi-models", + "anemoi-transform", + "anemoi-utils", "eccodes==2.38.3", "eccodes-cosmo-resources-python" ] @@ -24,5 +23,5 @@ torchaudio = { index = "pytorch" } [[tool.uv.index]] name = "pytorch" -url = "https://download.pytorch.org/whl/cu121" # need to couple torch with the correct cuda driver (cudatoolkit version) otherwise torch will be build but it won't be able to access GPUs. Won't throw any errors though! +url = "https://download.pytorch.org/whl/cu124" # need to couple torch with the correct cuda driver (cudatoolkit version) otherwise torch will be build but it won't be able to access GPUs. Won't throw any errors though! explicit = true From 07163ae0476f205ea2c13e2e1b309dabfecf93c8 Mon Sep 17 00:00:00 2001 From: Francesco Zanetta Date: Wed, 10 Dec 2025 10:45:30 +0100 Subject: [PATCH 2/3] change resolution order --- workflow/scripts/set_inference_pyproject.py | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/workflow/scripts/set_inference_pyproject.py b/workflow/scripts/set_inference_pyproject.py index 62cc8f3..4f99d72 100644 --- a/workflow/scripts/set_inference_pyproject.py +++ b/workflow/scripts/set_inference_pyproject.py @@ -33,36 +33,37 @@ "torch-geometric", ] +# NOTE: the order matters! First look for ECMWF repos, then MeteoSwiss ones _GIT_DEPENDENCIES_CONFIG = { "anemoi-models": { - "meteoswiss": { - "url": "https://github.com/MeteoSwiss/anemoi-core.git", - "subdirectory": "models", - }, "ecmwf": { "url": "https://github.com/ecmwf/anemoi-core.git", "subdirectory": "models", }, - }, - "anemoi-graphs": { "meteoswiss": { "url": "https://github.com/MeteoSwiss/anemoi-core.git", - "subdirectory": "graphs", + "subdirectory": "models", }, + }, + "anemoi-graphs": { "ecmwf": { "url": "https://github.com/ecmwf/anemoi-core.git", "subdirectory": "graphs", }, - }, - "anemoi-training": { "meteoswiss": { "url": "https://github.com/MeteoSwiss/anemoi-core.git", - "subdirectory": "training", + "subdirectory": "graphs", }, + }, + "anemoi-training": { "ecmwf": { "url": "https://github.com/ecmwf/anemoi-core.git", "subdirectory": "training", }, + "meteoswiss": { + "url": "https://github.com/MeteoSwiss/anemoi-core.git", + "subdirectory": "training", + }, }, "anemoi-datasets": { "ecmwf": {"url": "https://github.com/ecmwf/anemoi-datasets.git"} From e8d4e56a4425264b8eba81f05dcfad33e2c91738 Mon Sep 17 00:00:00 2001 From: Francesco Zanetta Date: Wed, 10 Dec 2025 11:01:08 +0100 Subject: [PATCH 3/3] remove anemoi-training from dependencies --- workflow/scripts/set_inference_pyproject.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/workflow/scripts/set_inference_pyproject.py b/workflow/scripts/set_inference_pyproject.py index 4f99d72..9891e09 100644 --- a/workflow/scripts/set_inference_pyproject.py +++ b/workflow/scripts/set_inference_pyproject.py @@ -55,16 +55,6 @@ "subdirectory": "graphs", }, }, - "anemoi-training": { - "ecmwf": { - "url": "https://github.com/ecmwf/anemoi-core.git", - "subdirectory": "training", - }, - "meteoswiss": { - "url": "https://github.com/MeteoSwiss/anemoi-core.git", - "subdirectory": "training", - }, - }, "anemoi-datasets": { "ecmwf": {"url": "https://github.com/ecmwf/anemoi-datasets.git"} },