fix unit test

apache · Oct 30, 2017 · cfa18af · cfa18af
1 parent 752b685
commit cfa18af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/clustering/PowerIterationClustering.scala b/mllib/src/main/scala/org/apache/spark/ml/clustering/PowerIterationClustering.scala
@@ -181,7 +181,7 @@ class PowerIterationClustering private[clustering] (
         require(nbr.size == weight.size,
           "The length of neighbor list must be equal to the the length of the weight list.")
         nbr.toArray.toIterator.zip(weight.toArray.toIterator)
-          .map(x => (id, x._1.toLong, x._2.toLong))}
+          .map(x => (id, x._1.toLong, x._2))}
     val algorithm = new MLlibPowerIterationClustering()
       .setK($(k))
       .setInitializationMode($(initMode))