torch<3,>=2.3
numpy<3,>=1.21.2
PyYAML<7,>=6

[all]
lightning<3,>=2.4
mlflow<4,>=3.11.1
scipy
scikit-learn
matplotlib

[develop]
pre-commit
ruff
pytest
pytest-cov
pytest-randomly
pytest-benchmark
mypy
types-PyYAML
types-setuptools

[lightning]
lightning<3,>=2.4

[mlflow]
mlflow<4,>=3.11.1

[plot]
matplotlib
scipy

[sampling]
scipy

[sklearn]
scikit-learn
