From 692df52ef801ce79180ada37dc0fbae94282ae34 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Mon, 27 Feb 2017 16:34:54 +0800
Subject: [PATCH] Fix unittest

---
 .travis.yml        | 2 +-
 python/setup.py.in | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 4fb2ca9387..3993dc31fd 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -57,7 +57,7 @@ before_install:
   - if [[ "$JOB" == "PRE_COMMIT" ]]; then sudo ln -s /usr/bin/clang-format-3.8 /usr/bin/clang-format; fi
   # Paddle is using protobuf 3.1 currently. Protobuf 3.2 breaks the compatibility. So we specify the python 
   # protobuf version.
-  - pip install numpy wheel 'protobuf==3.1' sphinx recommonmark sphinx_rtd_theme virtualenv pre-commit requests==2.9.2 LinkChecker
+  - pip install numpy wheel 'protobuf==3.1' sphinx recommonmark sphinx_rtd_theme virtualenv pre-commit requests==2.9.2 LinkChecker 'scikit-learn>=0.18.0'
 script:
   - paddle/scripts/travis/main.sh
 notifications:
diff --git a/python/setup.py.in b/python/setup.py.in
index 68ca35265c..fbee7bca90 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -15,5 +15,8 @@ setup(name='paddle',
       packages=packages,
       package_dir={
           '': '${CMAKE_CURRENT_SOURCE_DIR}'
-      }
+      },
+      install_requires = [
+          'scikit-learn>=0.18.0'
+      ]
 )
-- 
GitLab