diff --git a/graphx/src/main/scala/org/apache/spark/graphx/VertexRDD.scala b/graphx/src/main/scala/org/apache/spark/graphx/VertexRDD.scala index f1b6df9a3025e..f0f41b692fa86 100644 --- a/graphx/src/main/scala/org/apache/spark/graphx/VertexRDD.scala +++ b/graphx/src/main/scala/org/apache/spark/graphx/VertexRDD.scala @@ -111,6 +111,18 @@ class VertexRDD[@specialized VD: ClassTag]( partitionsRDD.map(_.size).reduce(_ + _) } + override def checkpoint() { + partitionsRDD.checkpoint() + } + + override def getCheckpointFile: Option[String] = { + partitionsRDD.getCheckpointFile + } + + override def isCheckpointed: Boolean = { + partitionsRDD.isCheckpointed + } + /** * Provides the `RDD[(VertexId, VD)]` equivalent output. */