Skip to content

Commit

Permalink
Minor update
Browse files Browse the repository at this point in the history
  • Loading branch information
jprakash-db committed Dec 19, 2024
1 parent 1ac441a commit fba9759
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
1 change: 0 additions & 1 deletion src/databricks/sql/auth/retry.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
4 changes: 2 additions & 2 deletions src/databricks/sql/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit fba9759

Please sign in to comment.