use input schema to get correct schema

delta-io · Aug 2, 2024 · ea8f068 · ea8f068
1 parent 28a4526
commit ea8f068
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/crates/core/src/operations/load_cdf.rs b/crates/core/src/operations/load_cdf.rs
@@ -230,11 +230,11 @@ impl CdfLoadBuilder {
         );
 
         let partition_values = self.snapshot.metadata().partition_columns.clone();
-        let schema = self.snapshot.arrow_schema()?;
+        let schema = self.snapshot.input_schema()?;
         let schema_fields: Vec<Field> = self
             .snapshot
-            .arrow_schema()?
-            .all_fields()
+            .input_schema()?
+            .flattened_fields()
             .into_iter()
             .filter(|f| !partition_values.contains(f.name()))
             .cloned()