From e2b0253d98573bea9f072c087b238a7ba30c4fd3 Mon Sep 17 00:00:00 2001 From: comphead Date: Tue, 16 Sep 2025 20:11:00 -0700 Subject: [PATCH] fix: correct `missingInput` for `CometHashAggregateExec` --- spark/src/main/scala/org/apache/spark/sql/comet/operators.scala | 2 ++ 1 file changed, 2 insertions(+) diff --git a/spark/src/main/scala/org/apache/spark/sql/comet/operators.scala b/spark/src/main/scala/org/apache/spark/sql/comet/operators.scala index 593f4f3a45..dd87328667 100644 --- a/spark/src/main/scala/org/apache/spark/sql/comet/operators.scala +++ b/spark/src/main/scala/org/apache/spark/sql/comet/operators.scala @@ -692,6 +692,8 @@ case class CometHashAggregateExec( override val serializedPlanOpt: SerializedPlan) extends CometUnaryExec with PartitioningPreservingUnaryExecNode { + override def producedAttributes: AttributeSet = outputSet ++ AttributeSet(resultExpressions) + override protected def withNewChildInternal(newChild: SparkPlan): SparkPlan = this.copy(child = newChild)