From fba9759e4a4073052d3bab0bcdf77b455c822189 Mon Sep 17 00:00:00 2001 From: Jothi Prakash Date: Thu, 19 Dec 2024 23:57:07 +0530 Subject: [PATCH] Minor update --- src/databricks/sql/auth/retry.py | 1 - src/databricks/sql/client.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/databricks/sql/auth/retry.py b/src/databricks/sql/auth/retry.py index 56cd3412..dcb48962 100755 --- a/src/databricks/sql/auth/retry.py +++ b/src/databricks/sql/auth/retry.py @@ -351,7 +351,6 @@ def should_retry(self, method: str, status_code: int) -> Tuple[bool, str]: Returns True if the request should be retried. Returns False or raises an exception if a retry would violate the configured policy. """ - # Request succeeded. Don't retry. if status_code == 200: return False, "200 codes are not retried" diff --git a/src/databricks/sql/client.py b/src/databricks/sql/client.py index 341a6ebe..aefed1ef 100755 --- a/src/databricks/sql/client.py +++ b/src/databricks/sql/client.py @@ -1225,7 +1225,7 @@ def __init__( self.description = execute_response.description self._arrow_schema_bytes = execute_response.arrow_schema_bytes self._next_row_index = 0 - self.use_cloud_fetch = use_cloud_fetch + self._use_cloud_fetch = use_cloud_fetch if execute_response.arrow_queue: # In this case the server has taken the fast path and returned an initial batch of @@ -1253,7 +1253,7 @@ def _fill_results_buffer(self): lz4_compressed=self.lz4_compressed, arrow_schema_bytes=self._arrow_schema_bytes, description=self.description, - use_cloud_fetch=self.use_cloud_fetch, + use_cloud_fetch=self._use_cloud_fetch, ) self.results = results self.has_more_rows = has_more_rows