# Demo requirements
# Versions unpinned to make it easier to test everything with
# the latest version
click

# Storage requirements
protobuf==3.20.1 # https://github.com/protocolbuffers/protobuf/issues/10051
google-cloud-storage
boto3
azure-core
azure-storage-blob
minio

# Data / dependencies for ML libraries
numpy>=1.19.2
python-Levenshtein==0.12.2
numba==0.55.1

# Prophet
pystan==2.19.1.1 # required to be installed before prophet

# Machine learning libraries
annoy
catboost
fastai
gensim
lightgbm
mxnet
onnx
onnxruntime
onnxmltools
prophet
pyspark
pytorch-lightning
scikit-learn
shap
skl2onnx
skorch
tensorflow
transformers
torch
torchvision
xgboost
