From 8e759dbb6b6f63ce9516833a1b024da1f6d63804 Mon Sep 17 00:00:00 2001 From: Jothi Prakash Date: Thu, 19 Dec 2024 10:52:04 +0530 Subject: [PATCH] Updated fix --- src/databricks/sql/thrift_backend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,