From 70abf9d08063d6b5a971761bdec907f89687c761 Mon Sep 17 00:00:00 2001 From: Peter Staar Date: Tue, 19 Nov 2024 05:59:07 +0100 Subject: [PATCH] fixed the mypy Signed-off-by: Peter Staar --- docling/backend/msexcel_backend.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docling/backend/msexcel_backend.py b/docling/backend/msexcel_backend.py index d35d6564..936c2ec4 100644 --- a/docling/backend/msexcel_backend.py +++ b/docling/backend/msexcel_backend.py @@ -1,7 +1,7 @@ import logging from io import BytesIO from pathlib import Path -from typing import Set, Tuple, Union +from typing import Dict, Set, Tuple, Union from docling_core.types.doc import ( DoclingDocument, @@ -51,7 +51,7 @@ class MsExcelDocumentBackend(DeclarativeDocumentBackend): # Initialise the parents for the hierarchy self.max_levels = 10 - self.parents = {} + self.parents: Dict[int, Any] = {} for i in range(-1, self.max_levels): self.parents[i] = None @@ -118,7 +118,7 @@ class MsExcelDocumentBackend(DeclarativeDocumentBackend): _log.info(f"Processing sheet: {sheet_name}") # Access the sheet by name - sheet = self.workbook[sheet_name] + sheet = self.workbook[sheet_name] self.parents[0] = doc.add_group( parent=None,