numpy
torch>=1.10.0
torchdrug
torch-scatter>=2.0.8
pandas<=1.3.5
tqdm
scikit-learn
class-resolver>=0.2.1
tabulate
pystow
pytdc
more-itertools

[docs]
sphinx
sphinx-rtd-theme
sphinx-click
sphinx-autodoc-typehints
sphinx_automodapi
nbsphinx_link
jupyter-sphinx

[tests]
pytest
pytest-cov
