dataclasses-json
numpy
click
click_default_group
sklearn
torch
tqdm
requests
optuna<1.2.0,>=1.0.0
pandas>=1.0.0
tabulate

[:python_version < "3.7"]
dataclasses

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

[mlflow]
mlflow>=1.8.0

[plotting]
matplotlib
seaborn

[templating]
jinja2
