chore: expose dataset field on TakeExec (#2425)

expose dataset field part of lancedb/sophon#1947 --------- Co-authored-by: Will Jones <[email protected]>
lancedb · May 31, 2024 · fbf3800 · fbf3800
1 parent 0429aa7
commit fbf3800
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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 {