Skip to content

Commit b507977

Browse files
authored
update load_from_doctags usage (#101)
Signed-off-by: Saidgurbuz <[email protected]>
1 parent 04fe2d9 commit b507977

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

docling_eval/evaluators/adapters.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,5 @@ def _doctags_to_doc(self, doc_id, doctags: str) -> DoclingDocument:
6060
"""
6161
doctags_page = DocTagsPage(tokens=doctags)
6262
doctags_doc = DocTagsDocument(pages=[doctags_page])
63-
doc = DoclingDocument(name=doc_id)
64-
doc.load_from_doctags(doctags_doc)
63+
doc = DoclingDocument.load_from_doctags(doctags_doc, document_name=doc_id)
6564
return doc

docling_eval/evaluators/base_evaluator.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,7 @@ def docling_document_from_doctags(
5656

5757
doctags_page = DocTagsPage(tokens=doctags, image=page_image)
5858
doctags_doc = DocTagsDocument(pages=[doctags_page])
59-
pred_doc = DoclingDocument(name=doc_id)
60-
pred_doc.load_from_doctags(doctags_doc)
59+
pred_doc = DoclingDocument.load_from_doctags(doctags_doc, document_name=doc_id)
6160

6261
return pred_doc
6362

docling_eval/prediction_providers/file_provider.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,8 +180,9 @@ def _load_doctags_doc(self, record: DatasetRecord) -> Optional[DoclingDocument]:
180180
# Build DoclingDocument
181181
doctags_page = DocTagsPage(tokens=doctags, image=page_image)
182182
doctags_doc = DocTagsDocument(pages=[doctags_page])
183-
doc = DoclingDocument(name=record.doc_id)
184-
doc.load_from_doctags(doctags_doc)
183+
doc = DoclingDocument.load_from_doctags(
184+
doctags_doc, document_name=record.doc_id
185+
)
185186

186187
return doc
187188
except Exception as e:

0 commit comments

Comments
 (0)