From 1b734ba045720e990124a90d55c57cd151f8ca45 Mon Sep 17 00:00:00 2001
From: SangBin Cho <rkooo567@gmail.com>
Date: Fri, 15 May 2020 16:54:54 -0700
Subject: [PATCH] Pin sklearn version (#8465)

---
 ci/travis/install-dependencies.sh | 2 +-
 docker/tune_test/requirements.txt | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/ci/travis/install-dependencies.sh b/ci/travis/install-dependencies.sh
index baeab4063..5467afbb1 100755
--- a/ci/travis/install-dependencies.sh
+++ b/ci/travis/install-dependencies.sh
@@ -212,7 +212,7 @@ install_dependencies() {
     pip_packages=(scipy tensorflow=="${TF_VERSION:-2.0.0b1}" cython==0.29.0 gym \
       opencv-python-headless pyyaml pandas==0.24.2 requests feather-format lxml openpyxl xlrd \
       py-spy pytest pytest-timeout networkx tabulate aiohttp uvicorn dataclasses pygments werkzeug \
-      kubernetes flask grpcio pytest-sugar pytest-rerunfailures pytest-asyncio scikit-learn numba \
+      kubernetes flask grpcio pytest-sugar pytest-rerunfailures pytest-asyncio scikit-learn==0.22.2 numba \
       Pillow prometheus_client)
     if [ "${OSTYPE}" != msys ]; then
       # These packages aren't Windows-compatible
diff --git a/docker/tune_test/requirements.txt b/docker/tune_test/requirements.txt
index afad51a7e..28479c4cb 100644
--- a/docker/tune_test/requirements.txt
+++ b/docker/tune_test/requirements.txt
@@ -18,6 +18,7 @@ opencv-python-headless
 pandas
 pytest-remotedata>=0.3.1
 pytest-timeout
+scikit-learn==0.22.2
 scikit-optimize
 sigopt
 smart_open