fix: Implement producedAttributes for CometWindowExec (#2789)

* Fix 2710 CometWindowExec should correctly account output attributes


Co-authored-by: Rahul Babarwal <rahul.babarwal@sumologic.com>

Author: rahulbabarwal89

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"),