Skip to content

Commit 458dd65

Browse files
committed
fix test eror
1 parent 0f3a11e commit 458dd65

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

python/pyspark/ml/classification.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1773,9 +1773,8 @@ class OneVsRest(Estimator, OneVsRestParams, HasParallelism, JavaMLReadable, Java
17731773
>>> model2 = OneVsRestModel.load(model_path)
17741774
>>> model2.transform(test0).head().prediction
17751775
0.0
1776-
>>> fieldNames = set(field.name for field in model.transform(test2).schema.fields)
1777-
>>> fieldNames
1778-
{'features', 'rawPrediction', 'prediction'}
1776+
>>> model.transform(test2).columns
1777+
['features', 'rawPrediction', 'prediction']
17791778
17801779
.. versionadded:: 2.0.0
17811780
"""

0 commit comments

Comments
 (0)