nit

apache · Nov 28, 2023 · 20ec741 · 20ec741
1 parent 28f4e87
commit 20ec741
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...-core/src/main/scala/org/apache/spark/sql/execution/ColumnarCollapseTransformStages.scala b/...-core/src/main/scala/org/apache/spark/sql/execution/ColumnarCollapseTransformStages.scala
@@ -176,7 +176,7 @@ case class ColumnarCollapseTransformStages(
   private def insertInputAdapter(plan: SparkPlan): SparkPlan = {
     plan match {
       case p if !supportTransform(p) =>
-        InputIteratorTransformer(new ColumnarInputAdapter(insertWholeStageTransformer(p)))
+        ColumnarCollapseTransformStages.wrapAdapter(insertWholeStageTransformer(p))
       case p =>
         p.withNewChildren(p.children.map(insertInputAdapter))
     }