Improve code

systemxlabs · Apr 29, 2024 · e2e39c5 · e2e39c5
1 parent f8b3118
commit e2e39c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ballista/core/src/serde/mod.rs b/ballista/core/src/serde/mod.rs
@@ -254,7 +254,7 @@ impl PhysicalExtensionCodec for BallistaPhysicalExtensionCodec {
                     protobuf::ShuffleReaderExecNode {
                         stage_id,
                         partition,
-                        schema: Some(exec.schema().as_ref().try_into()?),
+                        schema: Some(exec.schema.as_ref().try_into()?),
                     },
                 )),
             };
@@ -270,7 +270,7 @@ impl PhysicalExtensionCodec for BallistaPhysicalExtensionCodec {
                 physical_plan_type: Some(PhysicalPlanType::UnresolvedShuffle(
                     protobuf::UnresolvedShuffleExecNode {
                         stage_id: exec.stage_id as u32,
-                        schema: Some(exec.schema().as_ref().try_into()?),
+                        schema: Some(exec.schema.as_ref().try_into()?),
                         output_partition_count: exec.output_partition_count as u32,
                     },
                 )),