torch>=2
transformers>=4.28
lightning>=2
tables
torch_geometric
torch_scatter
numpy
more-itertools
lightning-cv[tune]>=1.0
optuna>=3.1
jsonargparse[signatures]>=4.35.1
einops
requests
tqdm
fair-esm==2.0.0
scikit-learn
attrs>=24.2
cattrs>=24.1

[:python_version < "3.10"]
typing-extensions>=4.11

[build]
ruff
mypy
isort
covdefaults

[dev]
ptn-set-transformer[build]
ptn-set-transformer[test]

[test]
pytest>=8.3
pytest-cov
