Updated fix

databricks · Dec 19, 2024 · 8e759db · 8e759db
1 parent b2202c7
commit 8e759db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,