Cleanup

Signed-off-by: Christoph Auer <[email protected]>
DS4SD · Sep 19, 2024 · d57c65d · d57c65d
1 parent 6ea6f29
commit d57c65d
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -16,7 +16,7 @@ ENV TORCH_HOME=/tmp/
 
 COPY examples/minimal.py /root/minimal.py
 
-#RUN python -c 'from deepsearch_glm.utils.load_pretrained_models import load_pretrained_nlp_models; load_pretrained_nlp_models(verbose=True);'
+RUN python -c 'from deepsearch_glm.utils.load_pretrained_models import load_pretrained_nlp_models; load_pretrained_nlp_models(verbose=True);'
 RUN python -c 'from docling.document_converter import DocumentConverter; artifacts_path = DocumentConverter.download_models_hf(force=True);'
 
 # On container environments, always set a thread budget to avoid undesired thread congestion.

diff --git a/docling/datamodel/document.py b/docling/datamodel/document.py
@@ -10,8 +10,8 @@
 from docling_core.types import PageDimensions, PageReference, Prov, Ref
 from docling_core.types import Table as DsSchemaTable
 from docling_core.types import TableCell
-from docling_core.types.legacy.base import BoundingBox as DsBoundingBox
-from docling_core.types.legacy.base import Figure
+from docling_core.types.doc.base import BoundingBox as DsBoundingBox
+from docling_core.types.doc.base import Figure
 from pydantic import BaseModel
 from typing_extensions import deprecated
 

diff --git a/tests/test_e2e_conversion.py b/tests/test_e2e_conversion.py
@@ -1,9 +1,8 @@
 from pathlib import Path
 
 from docling.backend.docling_parse_backend import DoclingParseDocumentBackend
-from docling.backend.pypdfium2_backend import PyPdfiumDocumentBackend
-from docling.datamodel.base_models import PipelineOptions
 from docling.datamodel.document import ConversionResult
+from docling.datamodel.pipeline_options import PipelineOptions
 from docling.document_converter import DocumentConverter
 
 from .verify_utils import verify_conversion_result

diff --git a/tests/test_interfaces.py b/tests/test_interfaces.py
@@ -5,8 +5,9 @@
 
 from docling.backend.docling_parse_backend import DoclingParseDocumentBackend
 from docling.backend.pypdfium2_backend import PyPdfiumDocumentBackend
-from docling.datamodel.base_models import DocumentStream, PipelineOptions
+from docling.datamodel.base_models import DocumentStream
 from docling.datamodel.document import ConversionResult, DocumentConversionInput
+from docling.datamodel.pipeline_options import PipelineOptions
 from docling.document_converter import DocumentConverter
 
 from .verify_utils import verify_conversion_result