Remove the test about whether "features" and "prediction" columns exi…

…st or not in Python
feynmanliang · Jul 17, 2015 · 4c61693 · 4c61693
1 parent fb2417c
commit 4c61693
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/python/pyspark/ml/clustering.py b/python/pyspark/ml/clustering.py
@@ -49,10 +49,6 @@ class KMeans(JavaEstimator, HasFeaturesCol, HasMaxIter, HasSeed):
     >>> len(centers)
     2
     >>> transformed = model.transform(df).select("features", "prediction")
-    >>> "features" in transformed.columns
-    True
-    >>> "prediction" in transformed.columns
-    True
     >>> rows = transformed.collect()
     >>> rows[0].prediction == rows[1].prediction
     True