From f822844a87ec67bfb5e138e891140be054581295 Mon Sep 17 00:00:00 2001 From: Christoph Auer Date: Tue, 29 Oct 2024 15:47:56 +0100 Subject: [PATCH] Optimize imports Signed-off-by: Christoph Auer --- docling/backend/asciidoc_backend.py | 4 ---- docling/datamodel/base_models.py | 2 +- docling/datamodel/document.py | 3 +-- docling/models/easyocr_model.py | 3 +-- docling/models/table_structure_model.py | 2 +- docling/pipeline/base_pipeline.py | 2 +- docling/pipeline/simple_pipeline.py | 2 +- docling/pipeline/standard_pdf_pipeline.py | 2 +- tests/test_backend_asciidoc.py | 3 --- tests/test_e2e_conversion.py | 3 --- 10 files changed, 7 insertions(+), 19 deletions(-) diff --git a/docling/backend/asciidoc_backend.py b/docling/backend/asciidoc_backend.py index c9d2fc52..829419af 100644 --- a/docling/backend/asciidoc_backend.py +++ b/docling/backend/asciidoc_backend.py @@ -1,24 +1,20 @@ import logging -import os import re from io import BytesIO from pathlib import Path from typing import Set, Union from docling_core.types.doc import ( - DocItem, DocItemLabel, DoclingDocument, DocumentOrigin, GroupItem, GroupLabel, ImageRef, - NodeItem, Size, TableCell, TableData, ) -from pydantic import AnyUrl from docling.backend.abstract_backend import DeclarativeDocumentBackend from docling.datamodel.base_models import InputFormat diff --git a/docling/datamodel/base_models.py b/docling/datamodel/base_models.py index a82d86a5..d06b6097 100644 --- a/docling/datamodel/base_models.py +++ b/docling/datamodel/base_models.py @@ -1,6 +1,6 @@ from enum import Enum, auto from io import BytesIO -from typing import TYPE_CHECKING, Dict, List, Optional, Set, Union +from typing import TYPE_CHECKING, Dict, List, Optional, Union from docling_core.types.doc import ( BoundingBox, diff --git a/docling/datamodel/document.py b/docling/datamodel/document.py index ef3ee85a..be4e9a12 100644 --- a/docling/datamodel/document.py +++ b/docling/datamodel/document.py @@ -1,6 +1,5 @@ import logging import re -import time from enum import Enum from io import BytesIO from pathlib import Path, PurePath @@ -53,7 +52,7 @@ from docling.datamodel.base_models import ( Page, ) from docling.datamodel.settings import DocumentLimits -from docling.utils.profiling import ProfilingItem, TimeRecorder +from docling.utils.profiling import ProfilingItem from docling.utils.utils import create_file_hash, create_hash if TYPE_CHECKING: diff --git a/docling/models/easyocr_model.py b/docling/models/easyocr_model.py index c767b8c9..8dff0fff 100644 --- a/docling/models/easyocr_model.py +++ b/docling/models/easyocr_model.py @@ -1,5 +1,4 @@ import logging -import time from typing import Iterable import numpy @@ -10,7 +9,7 @@ from docling.datamodel.document import ConversionResult from docling.datamodel.pipeline_options import EasyOcrOptions from docling.datamodel.settings import settings from docling.models.base_ocr_model import BaseOcrModel -from docling.utils.profiling import ProfilingItem, TimeRecorder +from docling.utils.profiling import TimeRecorder _log = logging.getLogger(__name__) diff --git a/docling/models/table_structure_model.py b/docling/models/table_structure_model.py index 5961c2c0..12bc2838 100644 --- a/docling/models/table_structure_model.py +++ b/docling/models/table_structure_model.py @@ -1,6 +1,6 @@ import copy from pathlib import Path -from typing import Iterable, List +from typing import Iterable import numpy from docling_core.types.doc import BoundingBox, DocItemLabel, TableCell diff --git a/docling/pipeline/base_pipeline.py b/docling/pipeline/base_pipeline.py index 6bfaea53..5013ad58 100644 --- a/docling/pipeline/base_pipeline.py +++ b/docling/pipeline/base_pipeline.py @@ -19,7 +19,7 @@ from docling.datamodel.document import ConversionResult, InputDocument from docling.datamodel.pipeline_options import PipelineOptions from docling.datamodel.settings import settings from docling.models.base_model import BaseEnrichmentModel -from docling.utils.profiling import ProfilingItem, ProfilingScope, TimeRecorder +from docling.utils.profiling import ProfilingScope, TimeRecorder from docling.utils.utils import chunkify _log = logging.getLogger(__name__) diff --git a/docling/pipeline/simple_pipeline.py b/docling/pipeline/simple_pipeline.py index f0108ab8..fb985231 100644 --- a/docling/pipeline/simple_pipeline.py +++ b/docling/pipeline/simple_pipeline.py @@ -5,7 +5,7 @@ from docling.backend.abstract_backend import ( DeclarativeDocumentBackend, ) from docling.datamodel.base_models import ConversionStatus -from docling.datamodel.document import ConversionResult, InputDocument +from docling.datamodel.document import ConversionResult from docling.datamodel.pipeline_options import PipelineOptions from docling.pipeline.base_pipeline import BasePipeline from docling.utils.profiling import ProfilingScope, TimeRecorder diff --git a/docling/pipeline/standard_pdf_pipeline.py b/docling/pipeline/standard_pdf_pipeline.py index 2def3fd5..65803d4f 100644 --- a/docling/pipeline/standard_pdf_pipeline.py +++ b/docling/pipeline/standard_pdf_pipeline.py @@ -7,7 +7,7 @@ from docling_core.types.doc import DocItem, ImageRef, PictureItem, TableItem from docling.backend.abstract_backend import AbstractDocumentBackend from docling.backend.pdf_backend import PdfDocumentBackend from docling.datamodel.base_models import AssembledUnit, Page -from docling.datamodel.document import ConversionResult, InputDocument +from docling.datamodel.document import ConversionResult from docling.datamodel.pipeline_options import ( EasyOcrOptions, PdfPipelineOptions, diff --git a/tests/test_backend_asciidoc.py b/tests/test_backend_asciidoc.py index ab94d58c..e4fae312 100644 --- a/tests/test_backend_asciidoc.py +++ b/tests/test_backend_asciidoc.py @@ -2,9 +2,6 @@ import glob import os from pathlib import Path -import pytest -from docling_core.types.doc import BoundingBox - from docling.backend.asciidoc_backend import AsciiDocBackend from docling.datamodel.base_models import InputFormat from docling.datamodel.document import InputDocument diff --git a/tests/test_e2e_conversion.py b/tests/test_e2e_conversion.py index 1e166116..c18a7a5b 100644 --- a/tests/test_e2e_conversion.py +++ b/tests/test_e2e_conversion.py @@ -1,8 +1,5 @@ from pathlib import Path -import yaml -from docling_core.types.doc import DoclingDocument - from docling.backend.docling_parse_backend import DoclingParseDocumentBackend from docling.datamodel.base_models import InputFormat from docling.datamodel.document import ConversionResult