torch>=1.6.0
numpy>=1.26
tqdm
scikit-learn
torchvision
rich
torch-geometric
wheel

[dev]
flake8>=6.1.0
black>=23.12.1
isort>=5.13.2
pre-commit>=3.6.0
pytest>=8.0.0
autopep8>=2.0.4
