add unit tests

2026-01-15 00:02:59 +00:00 · 2023-03-21 22:29:19 -07:00
parent 5c15e0ee86
commit 690141d357
5 changed files with 140 additions and 8 deletions
--- a/python/lancedb/query.py
+++ b/python/lancedb/query.py
@@ -78,7 +78,7 @@ class LanceQueryBuilder:
    def to_df(self) -> pd.DataFrame:
        """Execute the query and return the results as a pandas DataFrame.
        """
-        ds = self._table._dataset
+        ds = self._table.to_lance()
        # TODO indexed search
        tbl = ds.to_table(
            columns=self._columns,
@@ -89,4 +89,4 @@ class LanceQueryBuilder:
                "k": self._limit
            }
        )
-        return tbl.to_pandas()
+        return tbl.to_pandas()