anemoi-datasets>=0.4
anemoi-graphs
anemoi-models>=0.3
anemoi-utils[provenance]>=0.3.10
einops>=0.6.1
hydra-core>=1.3
matplotlib>=3.7.1
mlflow>=2.11.1
numpy<2
pynvml>=11.5
pyshtools>=4.10.4
pytorch-lightning>=2.1
timm>=0.9.2
torch>=2.2
torch-geometric<2.5,>=2.3.1
torchinfo>=1.8
tqdm>=4.65
zarr>=2.14.2

[all]

[dev]
anemoi-training[all,docs,tests]
pre-commit>=3.3.3

[docs]
nbsphinx
pandoc
sphinx
sphinx-argparse
sphinx-rtd-theme

[tests]
hypothesis
pytest
pytest-mock
