anemoi-datasets>=0.4
anemoi-graphs>=0.4
anemoi-models>=0.3
anemoi-utils[provenance]>=0.4.4
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

[profile]
holistictraceanalysis>=0.2
pandas>=1.3.2
rich>=13.6
tabulate>=0.9

[tests]
hypothesis
pytest
pytest-mock
