fix missing outputPartitioning in NativeParquetExec

kwai · Nov 6, 2023 · ff0f436 · ff0f436
1 parent ab3fd07
commit ff0f436
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...xtension/src/main/scala/org/apache/spark/sql/execution/blaze/plan/NativeProjectBase.scala b/...xtension/src/main/scala/org/apache/spark/sql/execution/blaze/plan/NativeProjectBase.scala
@@ -36,6 +36,7 @@ import org.blaze.protobuf.PhysicalExprNode
 import org.blaze.protobuf.PhysicalPlanNode
 import org.blaze.protobuf.ProjectionExecNode
 import org.apache.spark.sql.blaze.NativeSupports
+import org.apache.spark.sql.catalyst.plans.physical.Partitioning
 
 abstract class NativeProjectBase(
     projectList: Seq[NamedExpression],
@@ -58,6 +59,7 @@ abstract class NativeProjectBase(
       .toSeq: _*)
 
   override def output: Seq[Attribute] = projectList.map(_.toAttribute)
+  override def outputPartitioning: Partitioning = child.outputPartitioning
 
   private def nativeProject = getNativeProjectBuilder(projectList, addTypeCast).buildPartial()