numpy<2.0.0
pandas
torch
graphviz

[docs]
Sphinx>=3.0.0
docutils
pylons-sphinx-themes>=1.0.8
pylons_sphinx_latesturl
repoze.sphinx.autointerface
sphinxcontrib-autoprogram
sphinx-copybutton
sphinx-tabs
sphinx-panels
sphinx-rtd-theme
pillow>=6.2.0
openml
optuna
matplotlib
skorch

[mpi]
mpi4py>=3.1.2
