diff --git a/graph/src/main/scala/org/apache/spark/graph/Graph.scala b/graph/src/main/scala/org/apache/spark/graph/Graph.scala index 89e1b4ea01a81fa14a1a4d721853e871661b23f6..f0e96896de370ee024cac4f742a099cb15279644 100644 --- a/graph/src/main/scala/org/apache/spark/graph/Graph.scala +++ b/graph/src/main/scala/org/apache/spark/graph/Graph.scala @@ -3,6 +3,7 @@ package org.apache.spark.graph import org.apache.spark.rdd.RDD import org.apache.spark.util.ClosureCleaner +import org.apache.spark.Logging @@ -24,7 +25,7 @@ import org.apache.spark.util.ClosureCleaner * @tparam VD the vertex attribute type * @tparam ED the edge attribute type */ -abstract class Graph[VD: ClassManifest, ED: ClassManifest] { +abstract class Graph[VD: ClassManifest, ED: ClassManifest] extends Logging { /** * Get the vertices and their data. diff --git a/graph/src/main/scala/org/apache/spark/graph/GraphLab.scala b/graph/src/main/scala/org/apache/spark/graph/GraphLab.scala index b8503ab7fdb6c266161f31b3308bf7fcd554c1f8..39dc33acf0f6e954290f6bda1e3f5dba4f3b96e9 100644 --- a/graph/src/main/scala/org/apache/spark/graph/GraphLab.scala +++ b/graph/src/main/scala/org/apache/spark/graph/GraphLab.scala @@ -2,11 +2,12 @@ package org.apache.spark.graph import scala.collection.JavaConversions._ import org.apache.spark.rdd.RDD +import org.apache.spark.Logging /** * This object implements the GraphLab gather-apply-scatter api. */ -object GraphLab { +object GraphLab extends Logging { /** * Execute the GraphLab Gather-Apply-Scatter API 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 1750b7f8dcabc9a7d1b99bb4686df27a8624287e..c6b532462462fd14a7e2c0e1b250b664732b9a00 100644 --- a/graph/src/main/scala/org/apache/spark/graph/Pregel.scala +++ b/graph/src/main/scala/org/apache/spark/graph/Pregel.scala @@ -1,6 +1,7 @@ package org.apache.spark.graph import org.apache.spark.rdd.RDD +import org.apache.spark.Logging /** @@ -41,7 +42,7 @@ import org.apache.spark.rdd.RDD * }}} * */ -object Pregel { +object Pregel extends Logging { /**