chore: Code clean up

Signed-off-by: Nikos Livathinos <[email protected]>
DS4SD · Jan 13, 2025 · e53e1e5 · e53e1e5
1 parent 586ae01
commit e53e1e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docling_eval/evaluators/readingorder_evaluator.py b/docling_eval/evaluators/readingorder_evaluator.py
@@ -65,7 +65,7 @@ def __call__(
             # print(f"\n{i} - doc_id: {doc_id}")
             # self._show_items(true_doc)
 
-            reading_order = self._get_reading_order_preds(doc_id, true_doc)
+            reading_order = self._get_reading_order_preds(true_doc)
             if reading_order is None:
                 print(f"Broken input: {doc_id}")
                 broken_inputs += 1
@@ -97,7 +97,7 @@ def __call__(
 
         return ds_reading_order_evaluation
 
-    def _get_reading_order_preds(self, doc_id: str, true_doc: DoclingDocument):
+    def _get_reading_order_preds(self, true_doc: DoclingDocument):
         r""" """
         try:
             # Run the reading order model