Fixes + warning

Signed-off-by: Peter Ke <[email protected]>

Author: PeterKeDer

python/deltalake/_internal.pyi

@@ -875,7 +875,7 @@ class DeltaFileSystemHandler:
 
 class PyQueryBuilder:
     def __init__(self) -> None: ...
-    def register(self, table_name: str, delta_table: RawDeltaTable): ...
+    def register(self, table_name: str, delta_table: RawDeltaTable) -> None: ...
     def execute(self, sql: str) -> List[pyarrow.RecordBatch]: ...
 
 class DeltaDataChecker:

python/deltalake/query.py

@@ -1,15 +1,21 @@
 from __future__ import annotations
 
+import warnings
 from typing import List
 
 import pyarrow
 
 from deltalake._internal import PyQueryBuilder
 from deltalake.table import DeltaTable
+from deltalake.warnings import ExperimentalWarning
 
 
 class QueryBuilder:
     def __init__(self) -> None:
+        warnings.warn(
+            "QueryBuilder is experimental and subject to change",
+            category=ExperimentalWarning,
+        )
         self._query_builder = PyQueryBuilder()
 
     def register(self, table_name: str, delta_table: DeltaTable) -> QueryBuilder:

python/deltalake/warnings.py

@@ -0,0 +1,2 @@
+class ExperimentalWarning(Warning):
+    pass

python/src/query.rs

@@ -29,7 +29,6 @@ impl PyQueryBuilder {
         let log_store = delta_table._table.log_store();
 
         let scan_config = DeltaScanConfigBuilder::default()
-            .with_parquet_pushdown(false)
             .build(snapshot)
             .map_err(PythonError::from)?;