fix

viirya · Nov 23, 2024 · b1ae8fa · b1ae8fa
1 parent df545c5
commit b1ae8fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...rc/main/scala/org/apache/spark/sql/comet/execution/shuffle/CometShuffleExchangeExec.scala b/...rc/main/scala/org/apache/spark/sql/comet/execution/shuffle/CometShuffleExchangeExec.scala
@@ -500,7 +500,7 @@ class CometShuffleWriteProcessor(
       outputAttributes.length,
       nativePlan,
       nativeMetrics,
-      context.numPartitions(),
+      dep.rdd.getNumPartitions,
       context.partitionId())
 
     while (cometIter.hasNext) {