click<9.0.0,>=8.1.0
torch>=2.0.0
napistu>=0.8.0
numpy>=1.20.0
pandas>=1.3.0
pydantic>=2.0.0
rich<14.0.0,>=13.0.0
scikit-learn>=1.0.0

[analysis]
seaborn>=0.13.0

[dev]
black>=22.0.0
huggingface_hub[cli]>=0.20.0
lightning
pytest>=7.0.0
ruff>=0.1.0
seaborn>=0.13.0
torch-geometric>=2.4.0
torch-sparse>=0.6.17
torch-scatter>=2.1.1
wandb

[lightning]
huggingface_hub[cli]>=0.20.0
lightning
tqdm>=4.65.0
wandb

[pyg]
torch-geometric>=2.4.0
torch-sparse>=0.6.17
torch-scatter>=2.1.1
