feat(python): enable polars predict pushdown

2025-12-23 05:19:58 +00:00 · 2024-02-03 19:33:45 -08:00
2 changed files with 3 additions and 4 deletions
--- a/python/lancedb/table.py
+++ b/python/lancedb/table.py
@@ -836,9 +836,7 @@ class LanceTable(Table):
        -------
        pl.LazyFrame
        """
-        return pl.scan_pyarrow_dataset(
-            self.to_lance(), allow_pyarrow_filter=False, batch_size=batch_size
-        )
+        return pl.scan_pyarrow_dataset(self.to_lance(), batch_size=batch_size)

    @property
    def _dataset_uri(self) -> str:
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -14,7 +14,8 @@ dependencies = [
    "pyyaml>=6.0",
    "click>=8.1.7",
    "requests>=2.31.0",
-    "overrides>=0.7"
+    "overrides>=0.7",
+    "pyarrow>=14.0"
 ]
 description = "lancedb"
 authors = [{ name = "LanceDB Devs", email = "dev@lancedb.com" }]