Pin scikit-learn to version supporting Py2 (#195)

See tensorflow/data-validation#64 This commit may be reverted if that PR is merged.
spotify · May 10, 2019 · d6f6276 · d6f6276
1 parent e42f66c
commit d6f6276
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -3,6 +3,7 @@ numpy>=1.14.0,<2.0.0
 oauth2client<4.0.0
 pandas>=0.22.0,<1.0.0
 six>=1.11.0,<2.0.0
+scikit-learn>=0.18,<0.21
 tensorflow==1.13.1
 tensorflow-transform==0.13.0
 tensorflow-data-validation==0.13.1