Skip to content

Commit fb4967b

Browse files
committed
[SPARK-7432] [MLLIB] disable cv doctest
Temporarily disable flaky doctest for CrossValidator. jkbradley Author: Xiangrui Meng <[email protected]> Closes #5962 from mengxr/disable-pyspark-cv-test and squashes the following commits: 5db7e5b [Xiangrui Meng] disable cv doctest (cherry picked from commit 773aa25) Signed-off-by: Xiangrui Meng <[email protected]>
1 parent d6e76cb commit fb4967b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

python/pyspark/ml/tuning.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -103,10 +103,10 @@ class CrossValidator(Estimator):
103103
>>> grid = ParamGridBuilder().addGrid(lr.maxIter, [0, 1, 5]).build()
104104
>>> evaluator = BinaryClassificationEvaluator()
105105
>>> cv = CrossValidator(estimator=lr, estimatorParamMaps=grid, evaluator=evaluator)
106-
>>> cvModel = cv.fit(dataset)
107-
>>> expected = lr.fit(dataset, {lr.maxIter: 5}).transform(dataset)
108-
>>> cvModel.transform(dataset).collect() == expected.collect()
109-
True
106+
>>> # SPARK-7432: The following test is flaky.
107+
>>> # cvModel = cv.fit(dataset)
108+
>>> # expected = lr.fit(dataset, {lr.maxIter: 5}).transform(dataset)
109+
>>> # cvModel.transform(dataset).collect() == expected.collect()
110110
"""
111111

112112
# a placeholder to make it appear in the generated doc

0 commit comments

Comments
 (0)