From bafe673b97999d3023c946acd711ee9092aab64c Mon Sep 17 00:00:00 2001 From: Christoph Auer Date: Mon, 25 Nov 2024 14:36:56 +0100 Subject: [PATCH] fix: Other test fixes Signed-off-by: Christoph Auer --- tests/test_interfaces.py | 2 +- tests/test_invalid_input.py | 10 ++-------- 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/tests/test_interfaces.py b/tests/test_interfaces.py index a6675846..23bc3345 100644 --- a/tests/test_interfaces.py +++ b/tests/test_interfaces.py @@ -10,7 +10,7 @@ from docling.document_converter import DocumentConverter, PdfFormatOption from .verify_utils import verify_conversion_result_v1, verify_conversion_result_v2 -GENERATE = True +GENERATE = False def get_pdf_path(): diff --git a/tests/test_invalid_input.py b/tests/test_invalid_input.py index 0429612b..cbd12d09 100644 --- a/tests/test_invalid_input.py +++ b/tests/test_invalid_input.py @@ -3,14 +3,8 @@ from pathlib import Path import pytest -from docling.backend.docling_parse_backend import DoclingParseDocumentBackend -from docling.datamodel.base_models import ConversionStatus, DocumentStream, InputFormat -from docling.datamodel.pipeline_options import PdfPipelineOptions -from docling.document_converter import DocumentConverter, PdfFormatOption - -from .verify_utils import verify_conversion_result_v1, verify_conversion_result_v2 - -GENERATE = False +from docling.datamodel.base_models import ConversionStatus, DocumentStream +from docling.document_converter import DocumentConverter def get_pdf_path():