From 66ea3d13ad5d1f19adca60d30ab2e5c2eddb5110 Mon Sep 17 00:00:00 2001
From: Tim Saucer <timsaucer@gmail.com>
Date: Fri, 9 Jan 2026 09:09:38 -0500
Subject: [PATCH 1/3] Make OneShotPartitionStream pub

---
 rust/lance-datafusion/src/exec.rs      | 2 +-
 rust/lance/src/datafusion/dataframe.rs | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/rust/lance-datafusion/src/exec.rs b/rust/lance-datafusion/src/exec.rs
index 5a06ebf1de5..7a30562c147 100644
--- a/rust/lance-datafusion/src/exec.rs
+++ b/rust/lance-datafusion/src/exec.rs
@@ -732,7 +732,7 @@ pub trait SessionContextExt {
     ) -> datafusion::common::Result<DataFrame>;
 }
 
-struct OneShotPartitionStream {
+pub struct OneShotPartitionStream {
     data: Arc<Mutex<Option<SendableRecordBatchStream>>>,
     schema: Arc<ArrowSchema>,
 }
diff --git a/rust/lance/src/datafusion/dataframe.rs b/rust/lance/src/datafusion/dataframe.rs
index edb4ea05a68..8414d76a68e 100644
--- a/rust/lance/src/datafusion/dataframe.rs
+++ b/rust/lance/src/datafusion/dataframe.rs
@@ -182,7 +182,7 @@ pub trait SessionContextExt {
     ) -> datafusion::common::Result<DataFrame>;
 }
 
-struct OneShotPartitionStream {
+pub struct OneShotPartitionStream {
     data: Arc<Mutex<Option<SendableRecordBatchStream>>>,
     schema: Arc<Schema>,
 }

From c63d2b02a97fb3e254f91e5d211a40486d647cb8 Mon Sep 17 00:00:00 2001
From: Tim Saucer <timsaucer@gmail.com>
Date: Fri, 9 Jan 2026 09:25:59 -0500
Subject: [PATCH 2/3] Make new accessible

---
 rust/lance-datafusion/src/exec.rs      | 2 +-
 rust/lance/src/datafusion/dataframe.rs | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/rust/lance-datafusion/src/exec.rs b/rust/lance-datafusion/src/exec.rs
index 7a30562c147..b1600b8930a 100644
--- a/rust/lance-datafusion/src/exec.rs
+++ b/rust/lance-datafusion/src/exec.rs
@@ -748,7 +748,7 @@ impl std::fmt::Debug for OneShotPartitionStream {
 }
 
 impl OneShotPartitionStream {
-    fn new(data: SendableRecordBatchStream) -> Self {
+    pub fn new(data: SendableRecordBatchStream) -> Self {
         let schema = data.schema();
         Self {
             data: Arc::new(Mutex::new(Some(data))),
diff --git a/rust/lance/src/datafusion/dataframe.rs b/rust/lance/src/datafusion/dataframe.rs
index 8414d76a68e..76c4ac99e97 100644
--- a/rust/lance/src/datafusion/dataframe.rs
+++ b/rust/lance/src/datafusion/dataframe.rs
@@ -188,7 +188,7 @@ pub struct OneShotPartitionStream {
 }
 
 impl OneShotPartitionStream {
-    fn new(data: SendableRecordBatchStream) -> Self {
+    pub fn new(data: SendableRecordBatchStream) -> Self {
         let schema = data.schema();
         Self {
             data: Arc::new(Mutex::new(Some(data))),

From 59dfb486a0b57b939aa25fb370c3dd4c0c8dd0c5 Mon Sep 17 00:00:00 2001
From: Tim Saucer <timsaucer@gmail.com>
Date: Fri, 9 Jan 2026 09:26:18 -0500
Subject: [PATCH 3/3] Minor update to cargo lock

---
 python/Cargo.lock | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/Cargo.lock b/python/Cargo.lock
index 7b73a36aa6a..555308d942e 100644
--- a/python/Cargo.lock
+++ b/python/Cargo.lock
@@ -4557,6 +4557,7 @@ dependencies = [
  "roaring",
  "serde",
  "serde_json",
+ "smallvec",
  "snafu",
  "tantivy",
  "tempfile",