diff --git a/pyproject.toml b/pyproject.toml index e795dfc58..071c466a0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,7 +29,7 @@ numpy = [ sqlalchemy = "^1.3.24" openpyxl = "^3.0.10" alembic = "^1.0.11" -urllib3 = "^2.0.0" +urllib3 = ">=1.0" [tool.poetry.dev-dependencies] pytest = "^7.1.2" diff --git a/src/databricks/sql/auth/retry.py b/src/databricks/sql/auth/retry.py index 182c7137b..548b3622c 100644 --- a/src/databricks/sql/auth/retry.py +++ b/src/databricks/sql/auth/retry.py @@ -4,7 +4,13 @@ from enum import Enum from typing import List, Optional, Tuple, Union -from urllib3 import BaseHTTPResponse # type: ignore +# We only use this import for type hinting +try: + # If urllib3~=2.0 is installed + from urllib3 import BaseHTTPResponse # type: ignore +except ImportError: + # If urllib3~=1.0 is installed + from urllib3 import HTTPResponse as BaseHTTPResponse from urllib3 import Retry from urllib3.util.retry import RequestHistory