fix for migration test

lancedb · Nov 5, 2024 · c53fb4a · c53fb4a
1 parent 0d35fbc
commit c53fb4a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rust/lance/src/io/commit.rs b/rust/lance/src/io/commit.rs
@@ -350,6 +350,10 @@ fn fix_schema(manifest: &mut Manifest) -> Result<()> {
     for (old_field_id, new_field_id) in &old_field_id_mapping {
         let field = manifest.schema.mut_field_by_id(*old_field_id).unwrap();
         field.id = *new_field_id;
+
+        if let Some(local_field) = manifest.local_schema.mut_field_by_id(*old_field_id) {
+            local_field.id = *new_field_id;
+        }
     }
 
     // Drop data files that are no longer in use.