Skip to content

Commit 1b4154c

Browse files
jprakash-dbvarun-edachali-dbx
authored andcommitted
Added urllib3 version check (#547)
* Added version check * Removed packaging Signed-off-by: varun-edachali-dbx <[email protected]>
1 parent d403fcd commit 1b4154c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/databricks/sql/auth/retry.py

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

2221

2322

@@ -314,7 +313,9 @@ def get_backoff_time(self) -> float:
314313

315314
current_attempt = self.stop_after_attempts_count - int(self.total or 0)
316315
proposed_backoff = (2**current_attempt) * self.delay_min
317-
if version.parse(urllib3.__version__) >= version.parse("2.0.0"):
316+
317+
library_version = version("urllib3")
318+
if int(library_version.split(".")[0]) >= 2:
318319
if self.backoff_jitter != 0.0:
319320
proposed_backoff += random.random() * self.backoff_jitter
320321

0 commit comments

Comments
 (0)