From 440006666c5566279353d5bc264705c3b0d334f5 Mon Sep 17 00:00:00 2001
From: Beat Buesser <beat.buesser@ie.ibm.com>
Date: Sun, 15 Mar 2020 19:13:07 +0000
Subject: [PATCH] Update setup.py

Signed-off-by: Beat Buesser <beat.buesser@ie.ibm.com>
---
 setup.py | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/setup.py b/setup.py
index a91756c93a..150bad0d8c 100644
--- a/setup.py
+++ b/setup.py
@@ -11,24 +11,27 @@
                     'scipy',
                     'six',
                     'setuptools',
-                    'scikit-learn==0.22.1']
+                    'scikit-learn==0.22.1',
+                    'Pillow==7.0.0']
 
 # Comment out because of compatibility issues with numpy versions
 # 'catboost',
 tests_require = ['pytest-pep8',
-                 'keras>=2.2.5',
+                 'codecov'
                  'h5py',
-                 'mxnet',
-                 'Pillow',
                  'requests',
+                 'keras>=2.2.5',
+                 'mxnet',
                  'torch>=1.2.0',
-                 'tensorflow>=1.13.2',
+                 'tensorflow>=1.14.0',
                  'scikit-learn==0.22.1',
-                 'xgboost',
-                 'lightgbm',
-                 'GPy',
-                 'SciPy',
-                 'statsmodels']
+                 'xgboost==1.0.0',
+                 'lightgbm==2.3.1',
+                 'GPy==1.9.9',
+                 'numpy==1.18.1'
+                 'scipy==1.4.1',
+                 'statsmodels==0.11.0',
+                 'cma==2.7.0']
 
 docs_require = ['sphinx >= 1.4',
                 'sphinx_rtd_theme']