Use get, instead of apply

feynmanliang · Jul 16, 2015 · 6566c8b · 6566c8b
1 parent 288e8d5
commit 6566c8b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala
@@ -106,8 +106,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext {
     transformed.columns.foreach { column =>
       Array("features", predictionColName).contains(column)
     }
-    val clusters = transformed.select(predictionColName)
-      .map(row => row.apply(0)).distinct().collect().toSet
+    val clusters = transformed.select(predictionColName).map(_.get(0)).distinct().collect().toSet
     assert(clusters.size === k)
     assert(clusters === Set(0, 1, 2, 3, 4))
   }