torchmetrics
scikit-learn
backports.strenum
python-dotenv
matplotlib

[dev]
black
flake8
isort
pre-commit
bandit
pytest
pytest-cov
mkdocs
mkdocs-material
mike

[docs]
mkdocs
mkdocs-material
mike

[test]
pytest
pytest-cov
