From 4c6169357e84bf823c1346c64e1e41e86774e562 Mon Sep 17 00:00:00 2001 From: Yu ISHIKAWA Date: Fri, 17 Jul 2015 14:36:19 +0900 Subject: [PATCH] Remove the test about whether "features" and "prediction" columns exist or not in Python --- python/pyspark/ml/clustering.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/python/pyspark/ml/clustering.py b/python/pyspark/ml/clustering.py index 20dd5dbae4087..b5e9b6549d9f1 100644 --- 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