Skip to content

Commit 2601129

Browse files
committed
Fix Map return type error not previously caught
1 parent ec65502 commit 2601129

File tree

1 file changed

+4
-2
lines changed
  • core/src/main/scala/org/apache/spark/rdd

1 file changed

+4
-2
lines changed

core/src/main/scala/org/apache/spark/rdd/RDD.scala

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ package org.apache.spark.rdd
2020
import java.util.Random
2121

2222
import scala.collection.Map
23-
import scala.collection.JavaConversions.mapAsScalaMap
2423
import scala.collection.mutable.ArrayBuffer
2524
import scala.reflect.{classTag, ClassTag}
2625

@@ -848,7 +847,10 @@ abstract class RDD[T: ClassTag](
848847
m1
849848
}
850849
val myResult = mapPartitions(countPartition).reduce(mergeMaps)
851-
myResult.asInstanceOf[java.util.Map[T, Long]] // Will be wrapped as a Scala mutable Map
850+
// Convert to a Scala mutable map
851+
val mutableResult = scala.collection.mutable.Map[T,Long]()
852+
myResult.foreach { case (k, v) => mutableResult.put(k, v) }
853+
mutableResult
852854
}
853855

854856
/**

0 commit comments

Comments
 (0)