Revert ShuffleWriterExec schema change

tustvold · tustvold · commit 4b480f0fe32f · 2022-05-03T16:07:22.000+01:00
diff --git a/ballista/rust/core/src/execution_plans/shuffle_writer.rs b/ballista/rust/core/src/execution_plans/shuffle_writer.rs
@@ -300,7 +300,7 @@ impl ExecutionPlan for ShuffleWriterExec {
     }
 
     fn schema(&self) -> SchemaRef {
-        result_schema()
+        self.plan.schema()
     }
 
     fn output_partitioning(&self) -> Partitioning {

Original file line number	Diff line number	Diff line change
`@@ -300,7 +300,7 @@ impl ExecutionPlan for ShuffleWriterExec {`
`300`	`300`	`}`
`301`	`301`
`302`	`302`	`fn schema(&self) -> SchemaRef {`
`303`		`- result_schema()`
	`303`	`+ self.plan.schema()`
`304`	`304`	`}`
`305`	`305`
`306`	`306`	`fn output_partitioning(&self) -> Partitioning {`