Align buffers in RecordBatch.from_pyarrow_bound

apache · Sep 27, 2024 · 64373a7 · 64373a7
1 parent 62825b2
commit 64373a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/arrow/src/pyarrow.rs b/arrow/src/pyarrow.rs
@@ -363,13 +363,14 @@ impl FromPyArrow for RecordBatch {
 
             let schema_ptr = unsafe { schema_capsule.reference::<FFI_ArrowSchema>() };
             let ffi_array = unsafe { FFI_ArrowArray::from_raw(array_capsule.pointer().cast()) };
-            let array_data = unsafe { ffi::from_ffi(ffi_array, schema_ptr) }.map_err(to_py_err)?;
+            let mut array_data = unsafe { ffi::from_ffi(ffi_array, schema_ptr) }.map_err(to_py_err)?;
             if !matches!(array_data.data_type(), DataType::Struct(_)) {
                 return Err(PyTypeError::new_err(
                     "Expected Struct type from __arrow_c_array.",
                 ));
             }
             let options = RecordBatchOptions::default().with_row_count(Some(array_data.len()));
+            array_data.align_buffers();
             let array = StructArray::from(array_data);
             // StructArray does not embed metadata from schema. We need to override
             // the output schema with the schema from the capsule.