lightning>=2.0
numpy
pandas
rdkit
scikit-learn
scipy
torch>=2.1
astartes[molecules]
ConfigArgParse

[dev]
black==23.*
bumpversion
autopep8
flake8
pytest
pytest-cov

[docs]
nbsphinx
sphinx
sphinx-argparse
sphinx-autobuild
sphinx-autoapi
sphinxcontrib-bibtex
sphinx-book-theme
nbsphinx-link
ipykernel
docutils<0.21

[hpopt]
ray[tune]
hyperopt

[notebooks]
ipykernel

[test]
pytest>=6.2
pytest-cov
