Skip to content
This repository was archived by the owner on Dec 4, 2019. It is now read-only.

Commit 41cf622

Browse files
committed
Add upper version bound on scikit-learn, tighten GridSearchCV doctest
1 parent cd2b91f commit 41cf622

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

python/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
# This file should list any python package dependencies.
2-
scikit-learn>=0.18.1
2+
scikit-learn>=0.18.1, <=0.19

python/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
"Programming Language :: Python",
2020
"Topic :: Scientific/Engineering",
2121
]
22-
INSTALL_REQUIRES = ["scikit-learn >= 0.18.1"]
22+
INSTALL_REQUIRES = ["scikit-learn >=0.18.1, <= 0.19"]
2323

2424
# Project root
2525
ROOT = os.path.abspath(os.getcwd() + "/")

python/spark_sklearn/grid_search.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ class GridSearchCV(BaseSearchCV):
127127
kernel='rbf', max_iter=-1, probability=False,
128128
random_state=None, shrinking=True, tol=...,
129129
verbose=False),
130-
fit_params=..., iid=..., n_jobs=1,
130+
fit_params={}, iid=..., n_jobs=1,
131131
param_grid=..., pre_dispatch=..., refit=...,
132132
scoring=..., verbose=...)
133133
>>> sorted(clf.cv_results_.keys())

0 commit comments

Comments
 (0)