From e60c52586b8b5e9a7c5a98f349af9e8db432a528 Mon Sep 17 00:00:00 2001 From: Peter Staar Date: Sat, 19 Oct 2024 06:23:35 +0200 Subject: [PATCH] fixed the mypy Signed-off-by: Peter Staar --- docling/backend/asciidoc_backend.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/docling/backend/asciidoc_backend.py b/docling/backend/asciidoc_backend.py index 8a6ca7df..ad9544bf 100644 --- a/docling/backend/asciidoc_backend.py +++ b/docling/backend/asciidoc_backend.py @@ -15,15 +15,14 @@ from docling_core.types.doc import ( from docling.backend.abstract_backend import DeclarativeDocumentBackend from docling.datamodel.base_models import InputFormat - -# from docling.datamodel.document import InputDocument +from docling.datamodel.document import InputDocument _log = logging.getLogger(__name__) class AsciidocBackend(DeclarativeDocumentBackend): - def __init__(self, in_doc: "InputDocument", path_or_stream: Union[BytesIO, Path]): + def __init__(self, in_doc: InputDocument, path_or_stream: Union[BytesIO, Path]): super().__init__(in_doc, path_or_stream) self.path_or_stream = path_or_stream @@ -76,8 +75,9 @@ class AsciidocBackend(DeclarativeDocumentBackend): """ content = "" - with open(self.path_or_stream, "r") as fr: - self.lines = fr.readlines() + if isinstance(self.path_or_stream, Path): + with open(self.path_or_stream.name, "r") as fr: + self.lines = fr.readlines() # self.lines = file_content.splitlines() @@ -91,7 +91,7 @@ class AsciidocBackend(DeclarativeDocumentBackend): # Title if self.is_title(line): item = self.parse_title(line) - doc.add_text(text=item["text"], label="title") + doc.add_text(text=item["text"], label=DocItemLabel.TITLE) # Section headers elif self.is_section_header(line): @@ -125,7 +125,7 @@ class AsciidocBackend(DeclarativeDocumentBackend): # Plain text elif line: item = self.parse_text(line) - doc.add_text(text=item["text"], label="text") + doc.add_text(text=item["text"], label=DocItemLabel.TEXT) if in_table and len(table_data) > 0: data = self.populate_table_as_grid(table_data)