small fix

Signed-off-by: Firestarman <[email protected]>
NVIDIA · Oct 12, 2024 · 384576a · 384576a
1 parent df32c7e
commit 384576a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/src/test/scala/com/nvidia/spark/rapids/GpuShuffleCoalesceReaderRetrySuite.scala b/tests/src/test/scala/com/nvidia/spark/rapids/GpuShuffleCoalesceReaderRetrySuite.scala
@@ -49,9 +49,9 @@ class GpuShuffleCoalesceReaderRetrySuite extends RmmSparkRetrySuiteBase {
     serializedOneIntColumnBatch(4))
 
   test("GpuShuffleCoalesceReader split-retry") {
-    val iter = closeOnExcept(serializedBatches) { _ =>
+    val iter = closeOnExcept(serializedBatches) { serBatches =>
       val reader = new GpuShuffleCoalesceReader(
-        Iterator(serializedBatches: _*),
+        Iterator(serBatches: _*),
         targetBatchSize = 390, // each is 64 due to padding, then total is 320 (=64x5)
         dataTypes = Array(IntegerType),
         metricsMap = Map.empty.withDefaultValue(NoopMetric))