Fix self inconsistency with Seq in RapidsShuffleIterator

Co-authored-by: xieshuaihu <[email protected]>
NVIDIA · Jun 28, 2024 · 3dbab3d · 3dbab3d
1 parent 5fa9758
commit 3dbab3d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...lugin/src/main/spark340/scala/com/nvidia/spark/rapids/shuffle/RapidsShuffleIterator.scala b/...lugin/src/main/spark340/scala/com/nvidia/spark/rapids/shuffle/RapidsShuffleIterator.scala
@@ -173,8 +173,8 @@ class RapidsShuffleIterator(
     val (local, remote) = blocksByAddress.partition(ba => ba._1.host == localHost)
 
     (local ++ remote).foreach {
-      case (blockManagerId: BlockManagerId, blockIds: Seq[(BlockId, Long, Int)]) => {
-        val shuffleRequestsMapIndex: Seq[BlockIdMapIndex] =
+      case (blockManagerId: BlockManagerId, blockIds: collection.Seq[(BlockId, Long, Int)]) => {
+        val shuffleRequestsMapIndex =
           blockIds.map { case (blockId, _, mapIndex) =>
             /**
              * [[ShuffleBlockBatchId]] is an internal optimization in Spark, which will likely
@@ -193,7 +193,7 @@ class RapidsShuffleIterator(
                 throw new IllegalArgumentException(
                   s"${blockId.getClass} $blockId is not currently supported")
             }
-          }
+          }.toSeq
 
         val client = try {
           transport.makeClient(blockManagerId)