make type annotation compatible with py3.8

Author: yuanjieding-db

databricks/sdk/config.py

@@ -150,7 +150,7 @@ class Config:
     files_ext_multipart_upload_default_part_size: int = 10 * 1024 * 1024  # 10 MiB
 
     # List of multipart upload part sizes that can be automatically selected
-    files_ext_multipart_upload_part_size_options: list[int] = [
+    files_ext_multipart_upload_part_size_options: list = [
         10 * 1024 * 1024,  # 10 MiB
         20 * 1024 * 1024,  # 20 MiB
         50 * 1024 * 1024,  # 50 MiB

tests/test_files.py

@@ -297,14 +297,14 @@ def request(
         else:
             raise FallbackToDownloadUsingFilesApi("method must be HEAD or GET")
 
-    def _handle_head_file(self, headers: dict[str, str], url: str) -> "MockFilesApiDownloadResponse":
+    def _handle_head_file(self, headers: dict, url: str) -> "MockFilesApiDownloadResponse":
         if "If-Unmodified-Since" in headers:
             assert headers["If-Unmodified-Since"] == self.last_modified
         resp = MockFilesApiDownloadResponse(self, 0, None, MockFilesApiDownloadRequest(url))
         resp.content = ""
         return resp
 
-    def _handle_get_file(self, headers: dict[str, str], url: str) -> "MockFilesApiDownloadResponse":
+    def _handle_get_file(self, headers: dict, url: str) -> "MockFilesApiDownloadResponse":
         offset = 0
         end_byte_offset = None
         if "Range" in headers:
@@ -2678,9 +2678,9 @@ def test_resumable_upload(config: Config, test_case: ResumableUploadTestCase) ->
 
 @dataclass
 class CreateDownloadUrlResponseTestCase:
-    data: dict[str, Any]
+    data: dict
     expected_parsed_url: Optional[str] = None
-    expected_parsed_headers: Optional[dict[str, str]] = None
+    expected_parsed_headers: Optional[dict] = None
     expected_exception: Optional[Type[BaseException]] = None
 
     def run(self) -> None:

tests/test_files_utils.py

@@ -14,7 +14,7 @@
 
 class Utils:
     @staticmethod
-    def parse_range_header(range_header: str, content_length: Optional[int] = None) -> tuple[int, int]:
+    def parse_range_header(range_header: str, content_length: Optional[int] = None) -> tuple:
         """
         Parses a Range header string and returns the start and end byte positions.
         Example input: "bytes=0-499"
@@ -54,7 +54,7 @@ def read(self, size: int = -1) -> bytes:
     def readline(self, size: int = -1) -> bytes:
         return self._stream.readline(size)
 
-    def readlines(self, size: int = -1) -> list[bytes]:
+    def readlines(self, size: int = -1) -> list:
         return self._stream.readlines(size)
 
     def readable(self) -> bool:
@@ -73,7 +73,7 @@ def tell(self) -> int:
 class ConcatenatedInputStreamTestCase(ABC):
 
     @abstractmethod
-    def generate(self) -> tuple[bytes, BinaryIO]:
+    def generate(self) -> tuple:
         pass
 
 
@@ -83,7 +83,7 @@ def __init__(self, head: bytes, tail: bytes, is_seekable: bool = True):
         self._tail = tail
         self._is_seekable = is_seekable
 
-    def generate(self) -> tuple[bytes, BinaryIO]:
+    def generate(self) -> tuple:
         """
         Generate a pair of:
         (a) implementation under test
@@ -135,7 +135,7 @@ def to_string(test_case) -> str:
 ]
 
 
-def verify(test_case: ConcatenatedInputStreamTestCase, apply: Callable[[BinaryIO], tuple[any, bool]]):
+def verify(test_case: ConcatenatedInputStreamTestCase, apply: Callable[[BinaryIO], tuple]):
     """
     This method applies given function iteratively to both implementation under test
     and reference implementation of the stream, and verifies the result on each step is identical.
@@ -216,7 +216,7 @@ def apply(buffer: BinaryIO):
     verify(test_case, apply)
 
 
-def seeks_to_string(seeks: [tuple[int, int]]):
+def seeks_to_string(seeks: list):
     ", ".join(list(map(lambda seek: f"Seek: offset={seek[0]}, whence={seek[1]}", seeks)))
 
 
@@ -238,7 +238,7 @@ def seeks_to_string(seeks: [tuple[int, int]]):
     ],
     ids=seeks_to_string,
 )
-def test_seek(config, test_case: ConcatenatedInputStreamTestCase, seeks: list[tuple[int, int]]):
+def test_seek(config, test_case: ConcatenatedInputStreamTestCase, seeks: list):
     def read_and_restore(buf: BinaryIO) -> bytes:
         pos = buf.tell()
         result = buf.read()