fix for the (dumb) MyPy type checker

Signed-off-by: Joan Fabrégat <j@fabreg.at>
This commit is contained in:
Joan Fabrégat 2025-04-08 18:17:15 +02:00
parent 5b6086a489
commit 2632feeb75
No known key found for this signature in database

View File

@ -2,7 +2,7 @@ import logging
import sys import sys
import warnings import warnings
from pathlib import Path from pathlib import Path
from typing import Optional from typing import Optional, cast
from docling_core.types.doc import DocItem, ImageRef, PictureItem, TableItem from docling_core.types.doc import DocItem, ImageRef, PictureItem, TableItem
@ -227,7 +227,8 @@ class StandardPdfPipeline(PaginatedPipeline):
): ):
page_ix = element.prov[0].page_no - 1 page_ix = element.prov[0].page_no - 1
page = next( page = next(
(p for p in conv_res.pages if p.page_no == page_ix), None (p for p in conv_res.pages if p.page_no == page_ix),
cast("Page", None),
) )
assert page is not None assert page is not None
assert page.size is not None assert page.size is not None