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