Skip to content

Commit 945c14a

Browse files
committed
Fix Black formatting for retry.py and utils.py
Signed-off-by: Shubham Dhal <shubham.dhal@databricks.com>
1 parent 8f9d84b commit 945c14a

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

src/databricks/sql/auth/retry.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -390,7 +390,10 @@ def should_retry(
390390
# server explicitly signals it's safe via a Retry-After header. This prevents
391391
# duplicate side effects for non-idempotent operations.
392392
if self.respect_server_retry_after_header and not has_retry_after:
393-
return (False, "respect_server_retry_after_header mode: no Retry-After header present")
393+
return (
394+
False,
395+
"respect_server_retry_after_header mode: no Retry-After header present",
396+
)
394397

395398
# Request failed with 404 and was a GetOperationStatus. This is not recoverable. Don't retry.
396399
if status_code == 404 and self.command_type == CommandType.GET_OPERATION_STATUS:

src/databricks/sql/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -919,7 +919,9 @@ def build_client_context(server_hostname: str, version: str, **kwargs):
919919
),
920920
retry_delay_default=kwargs.get("_retry_delay_default"),
921921
retry_dangerous_codes=kwargs.get("_retry_dangerous_codes"),
922-
respect_server_retry_after_header=kwargs.get("_respect_server_retry_after_header"),
922+
respect_server_retry_after_header=kwargs.get(
923+
"_respect_server_retry_after_header"
924+
),
923925
proxy_auth_method=kwargs.get("_proxy_auth_method"),
924926
pool_connections=kwargs.get("_pool_connections"),
925927
pool_maxsize=kwargs.get("_pool_maxsize"),

0 commit comments

Comments
 (0)