fix

apache · Oct 1, 2024 · a6b8030 · a6b8030
1 parent c60db4d
commit a6b8030
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala
@@ -1000,6 +1000,8 @@ class CometSparkSessionExtensions
         case agg: CometHashAggregateExec if agg.aggregateExpressions.forall(_.mode == Partial) =>
           Some(agg)
         case agg: HashAggregateExec if agg.aggregateExpressions.forall(_.mode == Partial) => None
+        case agg: ObjectHashAggregateExec if agg.aggregateExpressions.forall(_.mode == Partial) =>
+          None
         case a: AQEShuffleReadExec => findPartialAgg(a.child)
         case s: ShuffleQueryStageExec => findPartialAgg(s.plan)
       }.flatten