Skip to content

Commit ff0ec64

Browse files
shivam2680varun-edachali-dbx
authored andcommitted
Add version check for urllib3 in backoff calculation (#526)
Signed-off-by: Shivam Raj <shivam.raj@databricks.com> Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 512d37c commit ff0ec64

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/databricks/sql/auth/retry.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717
from urllib3 import HTTPResponse as BaseHTTPResponse
1818
from urllib3 import Retry
1919
from urllib3.util.retry import RequestHistory
20+
from packaging import version
21+
2022

2123

2224
from databricks.sql.exc import (
@@ -312,9 +314,7 @@ def get_backoff_time(self) -> float:
312314

313315
current_attempt = self.stop_after_attempts_count - int(self.total or 0)
314316
proposed_backoff = (2**current_attempt) * self.delay_min
315-
316-
library_version = version("urllib3")
317-
if int(library_version.split(".")[0]) >= 2:
317+
if version.parse(urllib3.__version__) >= version.parse("2.0.0"):
318318
if self.backoff_jitter != 0.0:
319319
proposed_backoff += random.random() * self.backoff_jitter
320320

0 commit comments

Comments
 (0)