Skip to content

Commit 8338961

Browse files
committed
fix GraphX EdgeRDD zipPartitions
1 parent ae58aea commit 8338961

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphx/src/main/scala/org/apache/spark/graphx/EdgeRDD.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ package org.apache.spark.graphx
1919

2020
import scala.reflect.{classTag, ClassTag}
2121

22-
import org.apache.spark.{OneToOneDependency, Partition, Partitioner, TaskContext}
22+
import org.apache.spark._
2323
import org.apache.spark.rdd.RDD
2424
import org.apache.spark.storage.StorageLevel
2525

@@ -47,7 +47,7 @@ class EdgeRDD[@specialized ED: ClassTag, VD: ClassTag](
4747
* partitioner that allows co-partitioning with `partitionsRDD`.
4848
*/
4949
override val partitioner =
50-
partitionsRDD.partitioner.orElse(Some(Partitioner.defaultPartitioner(partitionsRDD)))
50+
partitionsRDD.partitioner.orElse(Some(new HashPartitioner(partitionsRDD.partitions.size)))
5151

5252
override def compute(part: Partition, context: TaskContext): Iterator[Edge[ED]] = {
5353
val p = firstParent[(PartitionID, EdgePartition[ED, VD])].iterator(part, context)

0 commit comments

Comments
 (0)