nit

apache · Nov 8, 2023 · 9b32e16 · 9b32e16
1 parent 9b1f6aa
commit 9b32e16
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala b/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala
@@ -49,7 +49,9 @@ 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(projections.flatten.flatMap(_.references))
 
   override def metricsUpdater(): MetricsUpdater =
     BackendsApiManager.getMetricsApiInstance.genExpandTransformerMetricsUpdater(metrics)

diff --git a/gluten-core/src/main/scala/io/glutenproject/execution/SortExecTransformer.scala b/gluten-core/src/main/scala/io/glutenproject/execution/SortExecTransformer.scala
@@ -59,7 +59,7 @@ case class SortExecTransformer(
   override def requiredChildDistribution: Seq[Distribution] =
     if (global) OrderedDistribution(sortOrder) :: Nil else UnspecifiedDistribution :: Nil
 
-  def getRelWithoutProject(
+  def getRelNode(
       context: SubstraitContext,
       sortOrder: Seq[SortOrder],
       originalInputAttributes: Seq[Attribute],
@@ -92,16 +92,6 @@ case class SortExecTransformer(
     }
   }
 
-  def getRelNode(
-      context: SubstraitContext,
-      sortOrder: Seq[SortOrder],
-      originalInputAttributes: Seq[Attribute],
-      operatorId: Long,
-      input: RelNode,
-      validation: Boolean): RelNode = {
-    getRelWithoutProject(context, sortOrder, originalInputAttributes, operatorId, input, validation)
-  }
-
   override protected def doValidateInternal(): ValidationResult = {
     if (!BackendsApiManager.getSettings.supportSortExec()) {
       return ValidationResult.notOk("Current backend does not support sort")