diff --git a/src/databricks/sql/thrift_backend.py b/src/databricks/sql/thrift_backend.py index a1f9815d..70b29d32 100644 --- a/src/databricks/sql/thrift_backend.py +++ b/src/databricks/sql/thrift_backend.py @@ -156,8 +156,6 @@ def __init__( ) # Cloud fetch - self._use_cloud_fetch = kwargs.get("use_cloud_fetch", True) - self.max_download_threads = kwargs.get("max_download_threads", 10) self._ssl_options = ssl_options @@ -900,6 +898,8 @@ def execute_command( ): assert session_handle is not None + self._use_cloud_fetch = use_cloud_fetch + spark_arrow_types = ttypes.TSparkArrowTypes( timestampAsArrow=self._use_arrow_native_timestamps, decimalAsArrow=self._use_arrow_native_decimals,