fix: In case pred_dict is given to TableEvaluator remove the ".png", …

…".jpg" suffixes from doc_id before trying to match Signed-off-by: Nikos Livathinos <[email protected]>
DS4SD · Mar 6, 2025 · fa3eba6 · fa3eba6
1 parent 72ee9f2
commit fa3eba6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docling_eval/evaluators/table_evaluator.py b/docling_eval/evaluators/table_evaluator.py
@@ -141,6 +141,8 @@ def __call__(
                 pred_doc_dict = data[BenchMarkColumns.PREDICTION]
                 pred_doc = DoclingDocument.model_validate_json(pred_doc_dict)
             else:
+                if doc_id.endswith(".png") or doc_id.endswith(".jpg"):
+                    doc_id = doc_id[:-4]
                 if doc_id not in pred_dict:
                     _log.error("Missing pred_doc from dict argument for %s", doc_id)
                     continue