diff options
| author | Joseph E. Gonzalez <joseph.e.gonzalez@gmail.com> | 2013-12-31 15:19:01 -0800 |
|---|---|---|
| committer | Joseph E. Gonzalez <joseph.e.gonzalez@gmail.com> | 2013-12-31 15:19:01 -0800 |
| commit | 3d93d7339626bec62d514cd524e45084a22715ae (patch) | |
| tree | 3e13b8453a75eaabf5acca1c11f91ceb90c9e30b /graph/src | |
| parent | 32d6ae9d9c15bdb7937e7bd69a55afb21631adb6 (diff) | |
| download | spark-3d93d7339626bec62d514cd524e45084a22715ae.tar.gz spark-3d93d7339626bec62d514cd524e45084a22715ae.tar.bz2 spark-3d93d7339626bec62d514cd524e45084a22715ae.zip | |
Fixing the persist behavior of the Vertex and Edge RDDs to persist the current RDD and not the parent.
Diffstat (limited to 'graph/src')
| -rw-r--r-- | graph/src/main/scala/org/apache/spark/graph/EdgeRDD.scala | 2 | ||||
| -rw-r--r-- | graph/src/main/scala/org/apache/spark/graph/VertexRDD.scala | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/graph/src/main/scala/org/apache/spark/graph/EdgeRDD.scala b/graph/src/main/scala/org/apache/spark/graph/EdgeRDD.scala index 63858db2ef..b1640bf9ce 100644 --- a/graph/src/main/scala/org/apache/spark/graph/EdgeRDD.scala +++ b/graph/src/main/scala/org/apache/spark/graph/EdgeRDD.scala @@ -33,7 +33,7 @@ class EdgeRDD[@specialized ED: ClassManifest]( * Caching a VertexRDD causes the index and values to be cached separately. */ override def persist(newLevel: StorageLevel): EdgeRDD[ED] = { - partitionsRDD.persist(newLevel) + super.persist(newLevel) this } diff --git a/graph/src/main/scala/org/apache/spark/graph/VertexRDD.scala b/graph/src/main/scala/org/apache/spark/graph/VertexRDD.scala index fe0f0ae491..671cf496f8 100644 --- a/graph/src/main/scala/org/apache/spark/graph/VertexRDD.scala +++ b/graph/src/main/scala/org/apache/spark/graph/VertexRDD.scala @@ -86,7 +86,7 @@ class VertexRDD[@specialized VD: ClassManifest]( * Caching a VertexRDD causes the index and values to be cached separately. */ override def persist(newLevel: StorageLevel): VertexRDD[VD] = { - partitionsRDD.persist(newLevel) + super.persist(newLevel) this } |