Fix dependency for lightgbm and aix python server (kubeflow#1349)

* Fix cvxpy to 1.1.7 * Fix numpy dep as 1.20.0 breaks lgbserver * Turn back lightgbm test
magdalenakuhn17 · Feb 3, 2021 · 5a9d7ac · 5a9d7ac
1 parent 722fec1
commit 5a9d7ac
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/python/aixexplainer/setup.py b/python/aixexplainer/setup.py
@@ -35,7 +35,8 @@
         "argparse >= 1.4.0",
         "aix360 >= 0.1.0",
         "lime >= 0.1.1.37",
-        "nest_asyncio>=1.4.0"
+        "nest_asyncio>=1.4.0",
+        "cvxpy == 1.1.7"
     ],
     tests_require=tests_require,
     extras_require={'test': tests_require}

diff --git a/python/lgbserver/setup.py b/python/lgbserver/setup.py
@@ -36,7 +36,8 @@
         "kfserving>=0.4.0",
         "lightgbm == 2.3.1",
         "pandas == 0.25.3",
-        "argparse >= 1.4.0"
+        "argparse >= 1.4.0",
+        "numpy == 1.19.5",
     ],
     tests_require=tests_require,
     extras_require={'test': tests_require}

diff --git a/test/scripts/run-e2e-tests.sh b/test/scripts/run-e2e-tests.sh
@@ -170,5 +170,5 @@ popd
 
 echo "Starting E2E functional tests ..."
 pushd test/e2e >/dev/null
-  pytest -n 3 --ignore=credentials/test_set_creds.py --ignore=predictor/test_lightgbm.py
+  pytest -n 3 --ignore=credentials/test_set_creds.py
 popd