This is an automated email from the ASF dual-hosted git repository.
comphead pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/datafusion-comet.git
The following commit(s) were added to refs/heads/main by this push:
new 71dda8ba2 fix: Implement producedAttributes for CometWindowExec (#2789)
71dda8ba2 is described below
commit 71dda8ba25b947cd16e920553df88c6551daf665
Author: rahulbabarwal89 <[email protected]>
AuthorDate: Mon Nov 17 22:55:12 2025 +0530
fix: Implement producedAttributes for CometWindowExec (#2789)
* Fix 2710 CometWindowExec should correctly account output attributes
Co-authored-by: Rahul Babarwal <[email protected]>
---
spark/src/main/scala/org/apache/spark/sql/comet/CometWindowExec.scala | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git
a/spark/src/main/scala/org/apache/spark/sql/comet/CometWindowExec.scala
b/spark/src/main/scala/org/apache/spark/sql/comet/CometWindowExec.scala
index 0a783b922..17f5c6246 100644
--- a/spark/src/main/scala/org/apache/spark/sql/comet/CometWindowExec.scala
+++ b/spark/src/main/scala/org/apache/spark/sql/comet/CometWindowExec.scala
@@ -21,7 +21,7 @@ package org.apache.spark.sql.comet
import scala.jdk.CollectionConverters._
-import org.apache.spark.sql.catalyst.expressions.{Alias, Attribute,
AttributeReference, CurrentRow, Expression, NamedExpression, RangeFrame,
RowFrame, SortOrder, SpecifiedWindowFrame, UnboundedFollowing,
UnboundedPreceding, WindowExpression}
+import org.apache.spark.sql.catalyst.expressions.{Alias, Attribute,
AttributeReference, AttributeSet, CurrentRow, Expression, NamedExpression,
RangeFrame, RowFrame, SortOrder, SpecifiedWindowFrame, UnboundedFollowing,
UnboundedPreceding, WindowExpression}
import
org.apache.spark.sql.catalyst.expressions.aggregate.{AggregateExpression,
Complete, Count, Max, Min, Sum}
import org.apache.spark.sql.catalyst.plans.physical.Partitioning
import org.apache.spark.sql.execution.SparkPlan
@@ -334,6 +334,7 @@ case class CometWindowExec(
extends CometUnaryExec {
override def nodeName: String = "CometWindowExec"
+ override def producedAttributes: AttributeSet = outputSet ++
AttributeSet(windowExpression)
override lazy val metrics: Map[String, SQLMetric] = Map(
"dataSize" -> SQLMetrics.createSizeMetric(sparkContext, "data size"),
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
