From e57e223ab7062f8d345768ba49b153cf4e0b5fc1 Mon Sep 17 00:00:00 2001 From: Christoph Auer Date: Mon, 23 Sep 2024 16:34:32 +0200 Subject: [PATCH] Fix formatting problems Signed-off-by: Christoph Auer --- docling/datamodel/document.py | 2 +- tests/verify_utils.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docling/datamodel/document.py b/docling/datamodel/document.py index fe0507b8..2f086b23 100644 --- a/docling/datamodel/document.py +++ b/docling/datamodel/document.py @@ -389,7 +389,7 @@ class ConvertedDocument(BaseModel): # table specific flags add_table_cell_location=add_table_cell_location, add_table_cell_label=add_table_cell_label, - add_table_cell_text=add_table_cell_text + add_table_cell_text=add_table_cell_text, ) def render_element_images( diff --git a/tests/verify_utils.py b/tests/verify_utils.py index 3dd94682..f75ed614 100644 --- a/tests/verify_utils.py +++ b/tests/verify_utils.py @@ -156,7 +156,7 @@ def verify_conversion_result( fw.write(doc_pred_md) with open(dt_path, "w") as fw: - fw.write(doc_pred_dt) + fw.write(doc_pred_dt) else: # default branch in test with open(pages_path, "r") as fr: doc_true_pages = PageList.validate_json(fr.read()) @@ -168,7 +168,7 @@ def verify_conversion_result( doc_true_md = fr.read() with open(dt_path, "r") as fr: - doc_true_dt = fr.read() + doc_true_dt = fr.read() assert verify_cells( doc_pred_pages, doc_true_pages @@ -188,4 +188,4 @@ def verify_conversion_result( assert verify_dt( doc_pred_dt, doc_true_dt - ), f"Mismatch in DocTags prediction for {input_path}" + ), f"Mismatch in DocTags prediction for {input_path}"