torch>=2.0.0
numpy<2.0

[cugraph]
pylibcugraphops-cu12

[dev]
turbo-gnn[test]

[dgl]
dgl==2.4.0

[full]
turbo-gnn[cugraph,notebook,test,tracking,viz]

[lint]
ruff==0.14.0
pre-commit>=3.0.0
mypy>=0.990
pylint>=2.15.0
isort>=5.11.0
types-PyYAML>=6.0.0
types-tqdm>=4.65.0

[notebook]
jupyterlab>=3.0.0
ipywidgets>=7.6.0
ipython>=8.0.0

[pyg]
torch-geometric
torch-scatter
torch-sparse
torch-cluster
torch-spline-conv

[research]
torch<2.5.0,>=2.4.0
numpy<2.0
scipy>=1.7.0
pandas>=1.5.0
scikit-learn>=1.0.0
pyyaml>=6.0
tqdm>=4.65.0
tensorboard>=2.10.0
psutil>=5.9.0
ogb>=1.3.0
torchdata==0.9.0
python-dotenv
nvidia-cutlass-dsl==4.3.0.dev0
tabulate

[test]
turbo-gnn[dgl,lint,pyg,research]
pytest>=7.0.0
pytest-cov>=4.0.0
pytest-xdist>=3.0.0

[tracking]
comet_ml>=3.53.2
python-dotenv

[viz]
matplotlib>=3.5.0
seaborn>=0.12.0
plotly>=5.0.0
