From 0bd92ed8d07712b7d8bb06378d877eb9643ba05a Mon Sep 17 00:00:00 2001 From: "Joseph E. Gonzalez" Date: Tue, 22 Oct 2013 19:10:51 -0700 Subject: [PATCH] Fixing a bug in pregel where the initial vertex-program results are lost. --- graph/src/main/scala/org/apache/spark/graph/Pregel.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graph/src/main/scala/org/apache/spark/graph/Pregel.scala b/graph/src/main/scala/org/apache/spark/graph/Pregel.scala index d1f5513f6a..2e3f86a3f0 100644 --- a/graph/src/main/scala/org/apache/spark/graph/Pregel.scala +++ b/graph/src/main/scala/org/apache/spark/graph/Pregel.scala @@ -50,7 +50,7 @@ object Pregel { def mapF(vid: Vid, edge: EdgeTriplet[VD,ED]) = sendMsg(edge.otherVertexId(vid), edge) // Receive the first set of messages - g.mapVertices( (vid, vdata) => vprog(vid, vdata, initialMsg)) + g = g.mapVertices( (vid, vdata) => vprog(vid, vdata, initialMsg)) while (i < numIter) { // compute the messages