torch>=1.6.0
numpy>=1.19.0
tqdm>=4.50.0
scikit-learn>=0.24.0
torchvision>=0.8.0
rich>=10.0.0

[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
sphinx>=4.0.0
sphinx-rtd-theme>=1.0.0

[geometric]
torch-geometric>=2.0.0
