From a4a316b2a38359b335f94b4574b4139302cee300 Mon Sep 17 00:00:00 2001
From: freeman <the.freeman.lab@gmail.com>
Date: Tue, 28 Oct 2014 22:13:06 -0700
Subject: [PATCH] Use collect

---
 .../org/apache/spark/mllib/clustering/StreamingKMeans.scala     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mllib/src/main/scala/org/apache/spark/mllib/clustering/StreamingKMeans.scala b/mllib/src/main/scala/org/apache/spark/mllib/clustering/StreamingKMeans.scala
index 9c898aa7c0664..0b07b70fa1101 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/clustering/StreamingKMeans.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/clustering/StreamingKMeans.scala
@@ -83,7 +83,7 @@ class StreamingKMeansModel(
       (p1._1 += p2._1, p1._2 + p2._2)
     }
     val pointStats: Array[(Int, (BV[Double], Long))] =
-      closest.reduceByKey{mergeContribs}.collectAsMap().toArray
+      closest.reduceByKey(mergeContribs).collect()
 
     // implement update rule
     for (newP <- pointStats) {