Cleanup

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

Hide file tree

Showing 2 changed files with 3 additions and 3 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