fix

apache · Nov 8, 2023 · 4a9d369 · 4a9d369
1 parent 9b32e16
commit 4a9d369
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gluten-core/src/main/scala/io/glutenproject/extension/PullOutPreProject.scala b/gluten-core/src/main/scala/io/glutenproject/extension/PullOutPreProject.scala
@@ -74,12 +74,12 @@ case class PullOutPreProject(spark: SparkSession) extends Rule[LogicalPlan] {
   }
 
   private def shouldAddPreProjectForAgg(agg: Aggregate): Boolean = {
-    agg.aggregateExpressions.exists(_.exists {
+    agg.aggregateExpressions.exists(_.find {
       case ae: AggregateExpression =>
         ae.aggregateFunction.children.exists(isNotAttributeAndLiteral) || ae.filter.exists(
           isNotAttributeAndLiteral)
       case _ => false
-    }) || agg.groupingExpressions.exists(e => isNotAttributeAndLiteral(e))
+    }.isDefined) || agg.groupingExpressions.exists(e => isNotAttributeAndLiteral(e))
   }
 
   private def shouldAddPreProjectForSort(sort: Sort): Boolean = {