From fbf3800ec5a3ffd219f9703c40b39b163223deba Mon Sep 17 00:00:00 2001
From: Rob Meng <rob.xu.meng@gmail.com>
Date: Fri, 31 May 2024 15:26:19 -0400
Subject: [PATCH] chore: expose dataset field on TakeExec (#2425)

expose dataset field

part of https://github.com/lancedb/sophon/issues/1947

---------

Co-authored-by: Will Jones <willjones127@gmail.com>
---
 rust/lance/src/io/exec/take.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/rust/lance/src/io/exec/take.rs b/rust/lance/src/io/exec/take.rs
index ea1a0194b1..d74ed61b1a 100644
--- a/rust/lance/src/io/exec/take.rs
+++ b/rust/lance/src/io/exec/take.rs
@@ -239,6 +239,13 @@ impl TakeExec {
             properties,
         })
     }
+
+    /// Get the dataset.
+    ///
+    /// WARNING: Internal API with no stability guarantees.
+    pub fn dataset(&self) -> &Arc<Dataset> {
+        &self.dataset
+    }
 }
 
 impl ExecutionPlan for TakeExec {