From 300d494d6e8409968b6fbb2d2fb425052966cfe0 Mon Sep 17 00:00:00 2001 From: Will Jones Date: Sat, 10 Oct 2026 17:12:05 +0000 Subject: [PATCH] docs: pass a RecordBatch in the merge_insert example Co-Authored-By: Claude Opus 5.5 --- rust/lancedb/src/table.rs | 34 ++++++++++++++-------------------- 1 file changed, 14 insertions(+), 20 deletions(-) diff --git a/rust/lancedb/src/table.rs b/rust/lancedb/src/table.rs index 17262ab3d..225ffaac0 100644 --- a/rust/lancedb/src/table.rs +++ b/rust/lancedb/src/table.rs @@ -1605,8 +1605,7 @@ impl Table { /// /// ```no_run /// # use std::sync::Arc; - /// # use arrow_array::{FixedSizeListArray, types::Float32Type, RecordBatch, - /// # RecordBatchIterator, Int32Array}; + /// # use arrow_array::{FixedSizeListArray, types::Float32Type, RecordBatch, Int32Array}; /// # use arrow_schema::{Schema, Field, DataType}; /// # tokio::runtime::Runtime::new().unwrap().block_on(async { /// let tmpdir = tempfile::tempdir().unwrap(); @@ -1620,30 +1619,25 @@ impl Table { /// # Field::new("vector", DataType::FixedSizeList( /// # Arc::new(Field::new("item", DataType::Float32, true)), 128), true), /// # ])); - /// let new_data = RecordBatchIterator::new( - /// vec![RecordBatch::try_new( - /// schema.clone(), - /// vec![ - /// Arc::new(Int32Array::from_iter_values(0..10)), - /// Arc::new( - /// FixedSizeListArray::from_iter_primitive::( - /// (0..10).map(|_| Some(vec![Some(1.0); 128])), - /// 128, - /// ), - /// ), - /// ], - /// ) - /// .unwrap()] - /// .into_iter() - /// .map(Ok), + /// let new_data = RecordBatch::try_new( /// schema.clone(), - /// ); + /// vec![ + /// Arc::new(Int32Array::from_iter_values(0..10)), + /// Arc::new( + /// FixedSizeListArray::from_iter_primitive::( + /// (0..10).map(|_| Some(vec![Some(1.0); 128])), + /// 128, + /// ), + /// ), + /// ], + /// ) + /// .unwrap(); /// // Perform an upsert operation /// let mut merge_insert = tbl.merge_insert(&["id"]); /// merge_insert /// .when_matched_update_all(None) /// .when_not_matched_insert_all(); - /// merge_insert.execute(Box::new(new_data) as Box).await.unwrap(); + /// merge_insert.execute(new_data).await.unwrap(); /// # }); /// ``` pub fn merge_insert(&self, on: &[&str]) -> MergeInsertBuilder {