torchmetrics>=0.6.0
tqdm
beartype
icontract

[:platform_machine != "x86_64" or sys_platform != "darwin"]
torch<=2.5.1,>=1.10.0
numpy<=2.2.4,>=1.22.0
scipy

[:platform_machine == "x86_64" and sys_platform == "darwin"]
torch<2.8.0,>=2.0.0
numpy<2.0.0,>=1.22.0
scipy<1.14.0

[docs]
mkdocs>=1.3
mkdocs-material
mkdocstrings-python
mkdocs-jupyter
mkdocs-literate-nav
mkdocs-gen-files
mkdocs-section-index

[tests]
pytest>=3.8
pytest-cov
hypothesis<6.136.0,>=6.105.0
pre-commit>=2.19
codespell
