Move shortest path and PageRank to bagel.examples

2 files changed, 7 insertions, 2 deletions

diff --git a/bagel/src/main/scala/bagel/ShortestPath.scala b/bagel/src/main/scala/bagel/examples/ShortestPath.scala
index 8f4a881850..2e6100c070 100644
--- a/bagel/src/main/scala/bagel/ShortestPath.scala
+++ b/bagel/src/main/scala/bagel/examples/ShortestPath.scala
@@ -1,10 +1,11 @@
-package bagel
+package bagel.examples
 
 import spark._
 import spark.SparkContext._
 
 import scala.math.min
 
+import bagel._
 import bagel.Pregel._
 
 object ShortestPath {
@@ -80,6 +81,7 @@ object ShortestPath {
   }
 }
 
+@serializable
 object MinCombiner extends Combiner[SPMessage, Int] {
   def createCombiner(msg: SPMessage): Int =
     msg.value
diff --git a/bagel/src/main/scala/bagel/WikipediaPageRank.scala b/bagel/src/main/scala/bagel/examples/WikipediaPageRank.scala
index 2fe77b4962..a5e0a9ffb6 100644
--- a/bagel/src/main/scala/bagel/WikipediaPageRank.scala
+++ b/bagel/src/main/scala/bagel/examples/WikipediaPageRank.scala
@@ -1,8 +1,9 @@
-package bagel
+package bagel.examples
 
 import spark._
 import spark.SparkContext._
 
+import bagel._
 import bagel.Pregel._
 
 import scala.collection.mutable.ArrayBuffer
@@ -75,6 +76,7 @@ object WikipediaPageRank {
   }
 }
 
+@serializable
 object PRCombiner extends Combiner[PRMessage, Double] {
   def createCombiner(msg: PRMessage): Double =
     msg.value
@@ -103,6 +105,7 @@ object PRCombiner extends Combiner[PRMessage, Double] {
   }
 }
 
+@serializable
 object PRNoCombiner extends DefaultCombiner[PRMessage] {
   def compute(numVertices: Long, epsilon: Double)(self: PRVertex, messages: Option[ArrayBuffer[PRMessage]], superstep: Int): (PRVertex, Iterable[PRMessage]) =
     PRCombiner.compute(numVertices, epsilon)(self, messages match {