shap~=0.39.0
scikit-learn>=0.23.0
pandas
torch
tb-nightly
seaborn
future
h5py
wandb
google-cloud
google-cloud-storage
pyyaml
plotly~=4.14.3
pytz~=2021.1
setuptools~=54.2.0
numpy~=1.20.2
requests~=2.25.1
torchvision>=0.6.0
mpld3~=0.3
numba>=0.50
sphinx
sphinx-rtd-theme
sphinx-autodoc-typehints

[dev]
autopep8
flake8
