diff --git a/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala b/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala index 245e40d6c42c..0a76d53a64e8 100644 --- 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)