Remove inline model stuff from tests

tests/builders/test_blank_page.py

@@ -5,9 +5,9 @@ from marker.builders.layout import LayoutBuilder
 from marker.builders.line import LineBuilder
 
 
-def test_blank_page(config, doc_provider, layout_model, ocr_error_model, recognition_model, detection_model, inline_detection_model):
+def test_blank_page(config, doc_provider, layout_model, ocr_error_model, detection_model):
     layout_builder = LayoutBuilder(layout_model, config)
-    line_builder = LineBuilder(detection_model, inline_detection_model, ocr_error_model)
+    line_builder = LineBuilder(detection_model, ocr_error_model)
     builder = DocumentBuilder(config)
     document = builder.build_document(doc_provider)
 

tests/builders/test_garbled_pdf.py

@@ -30,8 +30,8 @@ def test_garbled_pdf(pdf_document, detection_model, recognition_model, table_rec
 
 @pytest.mark.filename("hindi_judgement.pdf")
 @pytest.mark.config({"page_range": [2, 3], "disable_ocr": True})
-def test_garbled_builder(config, doc_provider, detection_model, inline_detection_model, ocr_error_model):
-    line_builder = LineBuilder(detection_model, inline_detection_model, ocr_error_model, config)
+def test_garbled_builder(config, doc_provider, detection_model, ocr_error_model):
+    line_builder = LineBuilder(detection_model, ocr_error_model, config)
     builder = DocumentBuilder(config)
     document = builder.build_document(doc_provider)
 
@@ -42,8 +42,8 @@ def test_garbled_builder(config, doc_provider, detection_model, inline_detection
 
 @pytest.mark.filename("adversarial.pdf")
 @pytest.mark.config({"page_range": [2, 3], "disable_ocr": True})
-def test_nongarbled_builder(config, doc_provider, detection_model, inline_detection_model, ocr_error_model):
-    line_builder = LineBuilder(detection_model, inline_detection_model, ocr_error_model, config)
+def test_nongarbled_builder(config, doc_provider, detection_model, ocr_error_model):
+    line_builder = LineBuilder(detection_model, ocr_error_model, config)
     builder = DocumentBuilder(config)
     document = builder.build_document(doc_provider)
 

tests/builders/test_inline_math_lines.py

@@ -3,6 +3,7 @@ import pytest
 from marker.processors.line_merge import LineMergeProcessor
 from marker.schema import BlockTypes
 
+@pytest.mark.skip("We do not support this functionality anymore")
 @pytest.mark.config({"page_range": [1]})
 def test_inline_box_nomerging(pdf_document, config):
     first_page = pdf_document.pages[0]
@@ -17,6 +18,7 @@ def test_inline_box_nomerging(pdf_document, config):
     assert line_count == 46
 
 
+@pytest.mark.skip("We do not support this functionality anymore")
 @pytest.mark.config({"page_range": [1], "use_llm": True})
 def test_inline_box_merging(pdf_document, config):
     first_page = pdf_document.pages[0]

tests/builders/test_layout_replace.py

@@ -9,10 +9,10 @@ from marker.schema.registry import get_block_class
 
 
 @pytest.mark.config({"page_range": [0]})
-def test_layout_replace(request, config, doc_provider, layout_model, ocr_error_model, detection_model, inline_detection_model):
+def test_layout_replace(request, config, doc_provider, layout_model, ocr_error_model, detection_model):
     # The llm layout builder replaces blocks - this makes sure text is still merged properly
     layout_builder = LayoutBuilder(layout_model, config)
-    line_builder = LineBuilder(detection_model, inline_detection_model, ocr_error_model, config)
+    line_builder = LineBuilder(detection_model, ocr_error_model, config)
     builder = DocumentBuilder(config)
     document = builder.build_document(doc_provider)
     layout_builder(document, doc_provider)

tests/conftest.py

@@ -54,10 +54,6 @@ def table_rec_model(model_dict):
 def ocr_error_model(model_dict):
     yield model_dict["ocr_error_model"]
 
-@pytest.fixture(scope="session")
-def inline_detection_model(model_dict):
-    yield model_dict["inline_detection_model"]
-
 @pytest.fixture(scope="function")
 def config(request):
     config_mark = request.node.get_closest_marker("config")
@@ -93,9 +89,9 @@ def doc_provider(request, config, temp_doc):
     yield provider_cls(temp_doc.name, config)
 
 @pytest.fixture(scope="function")
-def pdf_document(request, config, doc_provider, layout_model, ocr_error_model, recognition_model, detection_model, inline_detection_model):
+def pdf_document(request, config, doc_provider, layout_model, ocr_error_model, recognition_model, detection_model):
     layout_builder = LayoutBuilder(layout_model, config)
-    line_builder = LineBuilder(detection_model, inline_detection_model, ocr_error_model, config)
+    line_builder = LineBuilder(detection_model, ocr_error_model, config)
     ocr_builder = OcrBuilder(recognition_model, config)
     builder = DocumentBuilder(config)
     document = builder(doc_provider, layout_builder, line_builder, ocr_builder)