chore: streamline kwarg handling

CHANGELOG.md

@@ -1,5 +1,6 @@
-## 0.7.10-dev1
+## 0.7.10-dev2
 
+* Handle kwargs explicitly when needed, suppress otherwise
 * fix: Reduce Chipper memory consumption on x86_64 cpus
 * fix: Skips ordering elements coming from Chipper
 * fix: After refactoring to introduce Chipper, annotate() weren't able to show text with extra info from elements, this is fixed now.

test_unstructured_inference/inference/test_layout.py

@@ -2,7 +2,7 @@
 import os.path
 import tempfile
 from functools import partial
-from unittest.mock import mock_open, patch
+from unittest.mock import ANY, mock_open, patch
 
 import numpy as np
 import pytest
@@ -675,6 +675,8 @@ def test_process_file_with_model_routing(monkeypatch, model_type, is_detection_m
             fixed_layouts=None,
             extract_tables=False,
             pdf_image_dpi=200,
+            extract_images_in_pdf=ANY,
+            image_output_dir_path=ANY,
         )
 
 

unstructured_inference/__version__.py

@@ -1 +1 @@
-__version__ = "0.7.10-dev1"  # pragma: no cover
+__version__ = "0.7.10-dev2"  # pragma: no cover

unstructured_inference/inference/layout.py

@@ -71,10 +71,7 @@ def from_pages(cls, pages: List[PageLayout]) -> DocumentLayout:
     def from_file(
         cls,
         filename: str,
-        detection_model: Optional[UnstructuredObjectDetectionModel] = None,
-        element_extraction_model: Optional[UnstructuredElementExtractionModel] = None,
         fixed_layouts: Optional[List[Optional[List[TextRegion]]]] = None,
-        extract_tables: bool = False,
         pdf_image_dpi: int = 200,
         **kwargs,
     ) -> DocumentLayout:
@@ -108,11 +105,8 @@ def from_file(
                         image,
                         number=i + 1,
                         document_filename=filename,
-                        detection_model=detection_model,
-                        element_extraction_model=element_extraction_model,
                         layout=layout,
                         fixed_layout=fixed_layout,
-                        extract_tables=extract_tables,
                         **kwargs,
                     )
                     pages.append(page)
@@ -453,10 +447,6 @@ def from_image(
 def process_data_with_model(
     data: BinaryIO,
     model_name: Optional[str],
-    is_image: bool = False,
-    fixed_layouts: Optional[List[Optional[List[TextRegion]]]] = None,
-    extract_tables: bool = False,
-    pdf_image_dpi: int = 200,
     **kwargs,
 ) -> DocumentLayout:
     """Processes pdf file in the form of a file handler (supporting a read method) into a
@@ -467,10 +457,6 @@ def process_data_with_model(
         layout = process_file_with_model(
             tmp_file.name,
             model_name,
-            is_image=is_image,
-            fixed_layouts=fixed_layouts,
-            extract_tables=extract_tables,
-            pdf_image_dpi=pdf_image_dpi,
             **kwargs,
         )
 
@@ -484,6 +470,8 @@ def process_file_with_model(
     fixed_layouts: Optional[List[Optional[List[TextRegion]]]] = None,
     extract_tables: bool = False,
     pdf_image_dpi: int = 200,
+    extract_images_in_pdf: bool = False,
+    image_output_dir_path: Optional[str] = None,
     **kwargs,
 ) -> DocumentLayout:
     """Processes pdf file with name filename into a DocumentLayout by using a model identified by
@@ -514,6 +502,8 @@ def process_file_with_model(
             fixed_layouts=fixed_layouts,
             extract_tables=extract_tables,
             pdf_image_dpi=pdf_image_dpi,
+            extract_images_in_pdf=extract_images_in_pdf,
+            image_output_dir_path=image_output_dir_path,
             **kwargs,
         )
     )