Skip to content

Commit 7fecb86

Browse files
committed
Added version check
1 parent f9b7f43 commit 7fecb86

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/databricks/sql/auth/retry.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import random
33
import time
44
import typing
5+
from importlib.metadata import version
56
from enum import Enum
67
from typing import List, Optional, Tuple, Union
78

@@ -312,7 +313,9 @@ def get_backoff_time(self) -> float:
312313

313314
current_attempt = self.stop_after_attempts_count - int(self.total or 0)
314315
proposed_backoff = (2**current_attempt) * self.delay_min
315-
if version.parse(urllib3.__version__) >= version.parse("2.0.0"):
316+
317+
library_version = version("urllib3")
318+
if int(library_version.split(".")[0]) >= 2:
316319
if self.backoff_jitter != 0.0:
317320
proposed_backoff += random.random() * self.backoff_jitter
318321

0 commit comments

Comments
 (0)