updated pyproject toml to support legacy and anemoi. Use pip install …

….[legacy] or .[anemoi]
metno · Jan 9, 2025 · eb90cef · eb90cef
1 parent c718709
commit eb90cef
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -30,18 +30,28 @@ classifiers = [
     ]
 dependencies = [
     "anemoi-datasets",
-    "anemoi-models",
-    "anemoi-training",
     "argparse",
     "gridpp",
     "numpy",
     "pytest",
     "pytorch-lightning>=2.1",
     "torch",
+    "torchinfo",
     "tqdm",
     "verif",
     "zarr<=2.17",
 ]
+
+# legacy version supports this branch from MetNo fork of ECMWF anemoi-models
+# contains some optimizations and fixes 
+optional-dependencies.legacy = [
+    "anemoi-models @ git+https://github.com/metno/anemoi-models.git@feature/fix-memory-inference-chunking",
+]
+
+optional-dependencies.anemoi = [
+    "anemoi-models",
+    "anemoi-training",
+]
 optional-dependencies.tests = [
     "black",
     "coverage",