nn-template-core>=0.2
pytorch-lightning
torchmetrics>=0.10
hydra-core>=1.2
scikit-learn
backports.strenum
rich
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
