Skip to content

Commit

Permalink
[SPARK-7432] [MLLIB] disable cv doctest
Browse files Browse the repository at this point in the history
Temporarily disable flaky doctest for CrossValidator. jkbradley

Author: Xiangrui Meng <meng@databricks.com>

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 <meng@databricks.com>
  • Loading branch information
mengxr committed May 7, 2015
1 parent d6e76cb commit fb4967b
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions python/pyspark/ml/tuning.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,10 +103,10 @@ class CrossValidator(Estimator):
>>> grid = ParamGridBuilder().addGrid(lr.maxIter, [0, 1, 5]).build()
>>> evaluator = BinaryClassificationEvaluator()
>>> cv = CrossValidator(estimator=lr, estimatorParamMaps=grid, evaluator=evaluator)
>>> cvModel = cv.fit(dataset)
>>> expected = lr.fit(dataset, {lr.maxIter: 5}).transform(dataset)
>>> cvModel.transform(dataset).collect() == expected.collect()
True
>>> # SPARK-7432: The following test is flaky.
>>> # cvModel = cv.fit(dataset)
>>> # expected = lr.fit(dataset, {lr.maxIter: 5}).transform(dataset)
>>> # cvModel.transform(dataset).collect() == expected.collect()
"""

# a placeholder to make it appear in the generated doc
Expand Down

0 comments on commit fb4967b

Please sign in to comment.