Skip to content

Commit d32a61f

Browse files
ran pre-commit
Signed-off-by: Peter Staar <taa@zurich.ibm.com>
1 parent 4f3d59b commit d32a61f

File tree

2 files changed

+16
-8
lines changed

2 files changed

+16
-8
lines changed

docling_parse/pdf_parser.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -427,15 +427,24 @@ def _to_page_geometry_from_decoder(self, page_dim) -> PdfPageGeometry:
427427
coord_origin=CoordOrigin.BOTTOMLEFT,
428428
)
429429
art_bbox_obj = BoundingBox(
430-
l=crop_bbox[0], b=crop_bbox[1], r=crop_bbox[2], t=crop_bbox[3],
430+
l=crop_bbox[0],
431+
b=crop_bbox[1],
432+
r=crop_bbox[2],
433+
t=crop_bbox[3],
431434
coord_origin=CoordOrigin.BOTTOMLEFT,
432435
)
433436
media_bbox_obj = BoundingBox(
434-
l=media_bbox[0], b=media_bbox[1], r=media_bbox[2], t=media_bbox[3],
437+
l=media_bbox[0],
438+
b=media_bbox[1],
439+
r=media_bbox[2],
440+
t=media_bbox[3],
435441
coord_origin=CoordOrigin.BOTTOMLEFT,
436442
)
437443
crop_bbox_obj = BoundingBox(
438-
l=crop_bbox[0], b=crop_bbox[1], r=crop_bbox[2], t=crop_bbox[3],
444+
l=crop_bbox[0],
445+
b=crop_bbox[1],
446+
r=crop_bbox[2],
447+
t=crop_bbox[3],
439448
coord_origin=CoordOrigin.BOTTOMLEFT,
440449
)
441450

@@ -564,9 +573,7 @@ def _to_widgets_from_decoder(self, widgets_container) -> List[PdfWidget]:
564573

565574
return result
566575

567-
def _to_hyperlinks_from_decoder(
568-
self, hyperlinks_container
569-
) -> List[PdfHyperlink]:
576+
def _to_hyperlinks_from_decoder(self, hyperlinks_container) -> List[PdfHyperlink]:
570577
"""Convert typed PdfHyperlinks container to list of PdfHyperlink objects."""
571578
result: List[PdfHyperlink] = []
572579

tests/test_parse.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ def save_as_json_rounded(page: SegmentedPdfPage, filename, indent=2, ndigits=3):
4646
with open(filename, "w", encoding="utf-8") as fw:
4747
json.dump(out, fw, indent=indent)
4848

49+
4950
GROUNDTRUTH_FOLDER = "tests/data/groundtruth/"
5051
REGRESSION_FOLDER = "tests/data/regression/*.pdf"
5152

@@ -322,8 +323,8 @@ def verify_hyperlinks(
322323
abs(true_rect[l][1] - pred_rect[l][1]) < eps
323324
), "abs(true_rect[l][1]-pred_rect[l][1])<eps"
324325

325-
assert (
326-
str(true_hyperlink.uri) == str(pred_hyperlink.uri)
326+
assert str(true_hyperlink.uri) == str(
327+
pred_hyperlink.uri
327328
), "true_hyperlink.uri == pred_hyperlink.uri"
328329

329330
return True

0 commit comments

Comments
 (0)