formatting and linting

Author: Coniferish

_test_unstructured_client/unit_utils.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import pathlib
 from typing import Any
 from unittest.mock import (
     ANY,
@@ -31,6 +32,12 @@
     "property_mock",
 )
 
+def sample_docs_path(file_name: str) -> str:
+    """Resolve the absolute-path to `file_name` in the sample-docs directory."""
+    sample_docs_dir = pathlib.Path(__file__).parent / "_sample_docs"
+    file_path = sample_docs_dir / file_name
+    return str(file_path.resolve())
+
 
 # ------------------------------------------------------------------------------------------------
 # MOCKING FIXTURES

src/unstructured_client/_hooks/custom/form_utils.py

@@ -2,7 +2,7 @@
 
 import logging
 from pathlib import Path
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Mapping
 from typing_extensions import TypeAlias
 
 from requests_toolbelt.multipart.decoder import MultipartDecoder  # type: ignore
@@ -14,7 +14,7 @@
     from typing import Union
 
 logger = logging.getLogger(UNSTRUCTURED_CLIENT_LOGGER_NAME)
-FormData: TypeAlias = "dict[str, Union[str, shared.Files, list[str]]]"
+FormData: TypeAlias = "Mapping[str, Union[str, shared.Files, list[str]]]"
 
 PARTITION_FORM_FILES_KEY = "files"
 PARTITION_FORM_SPLIT_PDF_PAGE_KEY = "split_pdf_page"

src/unstructured_client/_hooks/custom/split_pdf_hook.py

@@ -548,7 +548,7 @@ def _get_pdf_chunk_paths(
         return pdf_chunk_paths
 
     def _get_pdf_chunk_files(
-            self, pdf_chunks: list[Tuple[Path, int]]
+        self, pdf_chunks: list[Tuple[Path, int]]
     ) -> Generator[Tuple[BinaryIO, int], None, None]:
         """Yields the file objects for the given pdf chunk paths.
 
@@ -575,8 +575,7 @@ def _get_pdf_chunk_files(
                 raise
             yield pdf_chunk_file, offset
 
-    def _await_elements(
-            self, operation_id: str) -> Optional[list]:
+    def _await_elements(self, operation_id: str) -> Optional[list]:
         """
         Waits for the partition requests to complete and returns the flattened
         elements.