Skip to content

Commit fba9759

Browse files
committed
Minor update
1 parent 1ac441a commit fba9759

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/databricks/sql/auth/retry.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,6 @@ def should_retry(self, method: str, status_code: int) -> Tuple[bool, str]:
351351
Returns True if the request should be retried. Returns False or raises an exception
352352
if a retry would violate the configured policy.
353353
"""
354-
355354
# Request succeeded. Don't retry.
356355
if status_code == 200:
357356
return False, "200 codes are not retried"

src/databricks/sql/client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1225,7 +1225,7 @@ def __init__(
12251225
self.description = execute_response.description
12261226
self._arrow_schema_bytes = execute_response.arrow_schema_bytes
12271227
self._next_row_index = 0
1228-
self.use_cloud_fetch = use_cloud_fetch
1228+
self._use_cloud_fetch = use_cloud_fetch
12291229

12301230
if execute_response.arrow_queue:
12311231
# 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):
12531253
lz4_compressed=self.lz4_compressed,
12541254
arrow_schema_bytes=self._arrow_schema_bytes,
12551255
description=self.description,
1256-
use_cloud_fetch=self.use_cloud_fetch,
1256+
use_cloud_fetch=self._use_cloud_fetch,
12571257
)
12581258
self.results = results
12591259
self.has_more_rows = has_more_rows

0 commit comments

Comments
 (0)