fix computePreferredLocations signature to not depend on underlying i…

…mplementation
pdeyhim · Apr 2, 2014 · 17e2907 · 17e2907
1 parent ea9de65
commit 17e2907
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/main/scala/org/apache/spark/scheduler/InputFormatInfo.scala b/core/src/main/scala/org/apache/spark/scheduler/InputFormatInfo.scala
@@ -164,7 +164,7 @@ object InputFormatInfo {
 
     PS: I know the wording here is weird, hopefully it makes some sense !
   */
-  def computePreferredLocations(formats: Seq[InputFormatInfo]): HashMap[String, HashSet[SplitInfo]]
+  def computePreferredLocations(formats: Seq[InputFormatInfo]): Map[String, Set[SplitInfo]]
   = {
 
     val nodeToSplit = new HashMap[String, HashSet[SplitInfo]]
@@ -178,6 +178,6 @@ object InputFormatInfo {
       }
     }
 
-    nodeToSplit
+    nodeToSplit.map(entry => entry._1 -> entry._2.toSet).toMap
   }
 }