From 7fecb86cf8e8411869e03db0ae7c6df0cae6d2a5 Mon Sep 17 00:00:00 2001 From: Jothi Prakash Date: Thu, 17 Apr 2025 15:23:32 +0530 Subject: [PATCH 1/2] Added version check --- src/databricks/sql/auth/retry.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/databricks/sql/auth/retry.py b/src/databricks/sql/auth/retry.py index 988206cb..789d2e8d 100755 --- a/src/databricks/sql/auth/retry.py +++ b/src/databricks/sql/auth/retry.py @@ -2,6 +2,7 @@ import random import time import typing +from importlib.metadata import version from enum import Enum from typing import List, Optional, Tuple, Union @@ -312,7 +313,9 @@ def get_backoff_time(self) -> float: current_attempt = self.stop_after_attempts_count - int(self.total or 0) proposed_backoff = (2**current_attempt) * self.delay_min - if version.parse(urllib3.__version__) >= version.parse("2.0.0"): + + library_version = version("urllib3") + if int(library_version.split(".")[0]) >= 2: if self.backoff_jitter != 0.0: proposed_backoff += random.random() * self.backoff_jitter From 97c39857fbbf6778cc1d5528e5df562a8380463b Mon Sep 17 00:00:00 2001 From: Jothi Prakash Date: Thu, 17 Apr 2025 15:25:20 +0530 Subject: [PATCH 2/2] Removed packaging --- src/databricks/sql/auth/retry.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/databricks/sql/auth/retry.py b/src/databricks/sql/auth/retry.py index 789d2e8d..432ac687 100755 --- a/src/databricks/sql/auth/retry.py +++ b/src/databricks/sql/auth/retry.py @@ -17,7 +17,6 @@ from urllib3 import HTTPResponse as BaseHTTPResponse from urllib3 import Retry from urllib3.util.retry import RequestHistory -from packaging import version from databricks.sql.exc import (