add unit tests

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()

python/lancedb/table.py

@@ -108,7 +108,7 @@ class LanceTable:
         return LanceQueryBuilder(self, query)
 
     @classmethod
-    def create(cls, db, name, data, schema):
+    def create(cls, db, name, data, schema=None):
         tbl = LanceTable(db, name)
         data = _sanitize_data(data, schema)
         lance.write_dataset(data, tbl._dataset_uri, mode="create")
@@ -131,10 +131,8 @@ def _sanitize_schema(data: pa.Table, schema: pa.Schema = None) -> pa.Table:
             return data
         # cast the columns to the expected types
         data = data.combine_chunks()
-        return pa.Table.from_arrays([
-            data[name].cast(schema.field(name).type)
-            for name in schema.names
-        ], schema=schema)
+        return pa.Table.from_arrays([data[name] for name in schema.names],
+                                    schema=schema)
     # just check the vector column
     return _sanitize_vector_column(data, vector_column_name=VECTOR_COLUMN_NAME)