[CORE] Fix ExpandExecTransformer miss input (#3665)

apache · Nov 10, 2023 · 797770d · 797770d
1 parent bd483c9
commit 797770d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala b/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala
@@ -50,7 +50,10 @@ case class ExpandExecTransformer(
   @transient override lazy val metrics =
     BackendsApiManager.getMetricsApiInstance.genExpandTransformerMetrics(sparkContext)
 
-  val originalInputAttributes: Seq[Attribute] = child.output
+  @transient
+  override lazy val references: AttributeSet = {
+    AttributeSet.fromAttributeSets(projections.flatten.map(_.references))
+  }
 
   override def metricsUpdater(): MetricsUpdater =
     BackendsApiManager.getMetricsApiInstance.genExpandTransformerMetricsUpdater(metrics)