dataclasses-json
numpy
scipy>=1.7.0
click
click_default_group
scikit-learn
tqdm
requests
optuna>=2.0.0
pandas>=1.0.0
tabulate
more_click
more_itertools
pystow>=0.4.3
docdata
class_resolver>=0.3.10
pyyaml
rexmex>=0.1.3
torch_max_mem>=0.0.4
torch-ppr>=0.0.7
protobuf<4.0.0
typing_extensions

[:platform_system != "Windows"]
torch>=1.10

[biomedicine]
bioregistry
pyobo>=0.8.7

[docs]
sphinx
sphinx-rtd-theme
sphinx-click
sphinx-autodoc-typehints<=1.13.1
sphinx_automodapi
texext

[lightning]
pytorch_lightning>=1.7.2

[mlflow]
mlflow>=1.8.0

[neptune]
neptune-client

[ogb]
ogb>=1.3.4

[opt_einsum]
opt_einsum

[plotting]
matplotlib
seaborn

[templating]
jinja2

[tensorboard]
tensorboard

[tests]
unittest-templates>=0.0.5
coverage
pytest

[transformers]
transformers

[wandb]
wandb

[wordcloud]
wordcloud
IPython
