From aa9469d02e651e0f0d925877e32532869bfba067 Mon Sep 17 00:00:00 2001 From: Yu ISHIKAWA <yuu.ishikawa@gmail.com> Date: Sat, 13 Jun 2015 01:59:25 +0900 Subject: [PATCH] Fix a python test suite error caused by python 3.x --- python/pyspark/ml/clustering.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/pyspark/ml/clustering.py b/python/pyspark/ml/clustering.py index cc0dcc7b89430..b23f7b3c2d280 100644 --- a/python/pyspark/ml/clustering.py +++ b/python/pyspark/ml/clustering.py @@ -49,8 +49,10 @@ class KMeans(JavaEstimator, HasFeaturesCol, HasMaxIter, HasSeed): >>> len(centers) 2 >>> transformed = model.transform(df) - >>> transformed.columns - [u'features', u'prediction'] + >>> (transformed.columns)[0] == 'features' + True + >>> (transformed.columns)[1] == 'prediction' + True >>> rows = sorted(transformed.collect(), key = lambda r: r[0]) >>> rows[0].prediction == rows[1].prediction True