From 0bd92ed8d07712b7d8bb06378d877eb9643ba05a Mon Sep 17 00:00:00 2001
From: "Joseph E. Gonzalez" <joseph.e.gonzalez@gmail.com>
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
--
GitLab