fix wrong type

zhengchenyu · Dec 27, 2023 · 52464e1 · 52464e1
1 parent 88fd811
commit 52464e1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/src/main/scala/org/apache/spark/shuffle/BlockStoreShuffleReader.scala b/core/src/main/scala/org/apache/spark/shuffle/BlockStoreShuffleReader.scala
@@ -122,7 +122,9 @@ private[spark] class BlockStoreShuffleReader[K, C](
           aggregated = true
           if (dep.mapSideCombine) {
             new ExternalSorter[K, C, C](context,
-              dep.aggregator.asInstanceOf[Option[Aggregator[K, C, C]]],
+              Option(new Aggregator[K, C, C](identity,
+                dep.aggregator.get.mergeCombiners,
+                dep.aggregator.get.mergeCombiners)),
               ordering = Some(keyOrd), serializer = dep.serializer)
           } else {
             new ExternalSorter[K, Nothing, C](context,