Fix Python3 error

apache · Jun 18, 2015 · 8ab9e89 · 8ab9e89
1 parent a9817df
commit 8ab9e89
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/api/python/PythonMLLibAPI.scala b/mllib/src/main/scala/org/apache/spark/mllib/api/python/PythonMLLibAPI.scala
@@ -975,8 +975,7 @@ private[python] class PythonMLLibAPI extends Serializable {
       val model = new StreamingKMeansModel(
         clusterCenters.asScala.toArray, clusterWeights.asScala.toArray)
         .update(data, decayFactor, timeUnit)
-      List(model.clusterCenters, model.clusterWeights).
-        map(_.asInstanceOf[Object]).asJava
+      List[AnyRef](model.clusterCenters, Vectors.dense(model.clusterWeights)).asJava
   }
 
 }

diff --git a/python/pyspark/mllib/tests.py b/python/pyspark/mllib/tests.py
@@ -25,7 +25,8 @@
 import array as pyarray
 from time import time, sleep
 
-from numpy import array, array_equal, zeros, inf, all, random, sum
+from numpy import array, array_equal, zeros, inf, all, random
+from numpy import sum as array_sum
 from py4j.protocol import Py4JJavaError
 
 if sys.version_info[:2] <= (2, 6):
@@ -908,7 +909,7 @@ def test_accuracy_for_single_center(self):
         self.ssc.start()
         self._ssc_wait(t, 10.0, 0.01)
         self.assertEquals(stkm.latestModel.getClusterWeights, [25.0])
-        realCenters = sum(array(centers), axis=0)
+        realCenters = array_sum(array(centers), axis=0)
         for i in range(d):
             modelCenters = stkm.latestModel.centers[0][i]
             self.assertAlmostEqual(centers[0][i], modelCenters, 1)