mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-26 20:14:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use yield from correctly?

Browse Source 
Signed-off-by: Vinay Damodaran <vrdn@hey.com>
This commit is contained in:
Vinay Damodaran 2025-05-07 20:50:52 -07:00
parent e82bcaae61
commit 45c5a9445a
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docling/models/api_vlm_model.py
View File

@ -70,5 +70,4 @@ class ApiVlmModel(BasePageModel):
return page return page
with ThreadPoolExecutor(max_workers=concurrency) as executor: with ThreadPoolExecutor(max_workers=concurrency) as executor:
for result in executor.map(_vlm_request, page_batch): yield from executor.map(_vlm_request, page_batch)
yield from result

3
docling/models/picture_description_api_model.py
View File

@ -62,5 +62,4 @@ class PictureDescriptionApiModel(PictureDescriptionBaseModel):
) )
with ThreadPoolExecutor(max_workers=concurrency) as executor: with ThreadPoolExecutor(max_workers=concurrency) as executor:
for result in executor.map(_api_request, images): yield from executor.map(_api_request, images)
yield from result