requirements.txt

# base
numpy==1.18.1
scipy==1.4.1
matplotlib==3.2.1
scikit-learn==0.22.1
six==1.15.0
Pillow==7.2.0
tqdm==4.45.0
statsmodels==0.11.0
pydub==0.24.1
resampy==0.2.2
ffmpeg-python==0.2.0
cma==3.0.3

# frameworks
h5py==2.10.0
# supported versions: (tensorflow==2.2.0 with keras==2.3.1) or (tensorflow==1.15.3 with keras==2.2.5)
tensorflow
keras
tensorflow-addons==0.10.0
mxnet==1.6.0
--find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.5.0+cpu
torchvision==0.6.0+cpu
catboost==0.24
GPy==1.9.9
lightgbm==2.3.1
xgboost==1.1.1

# other
pytest==5.4.1
pytest-pep8==1.0.6
pytest-mock==3.2.0
codecov==2.0.22
requests==2.23.0

# ART
-e .

#NOTE to contributors: When changing/adding packages, please make sure that the packages are consitent with those
# present within the Dockerfile