shap==0.39.0
scikit-learn>=0.23.0
pandas
torch
tb-nightly
seaborn
future
h5py
wandb==0.12.2
google-cloud
google-cloud-storage
pyyaml
plotly~=5.1.0
pytz~=2021.1
setuptools~=58.0.0
numpy==1.20
requests
torchvision>=0.6.0
mpld3~=0.5
numba>=0.50
sphinx
sphinx-rtd-theme
sphinx-autodoc-typehints

[dev]
autopep8
flake8
