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

Unload doc backend

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2025-07-18 16:48:35 +02:00
parent 988db91bff
commit 01066f0b6e
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docling/pipeline/threaded_standard_pdf_pipeline.py
View File

@ -527,6 +527,9 @@ class ThreadedStandardPdfPipeline(BasePipeline):
for st in ctx.stages: for st in ctx.stages:
st.stop() st.stop()
ctx.output_queue.close() ctx.output_queue.close()
if conv_res.input._backend is not None:
conv_res.input._backend.unload()
return conv_res return conv_res
# -------------------------------------------------------------- feed_pages # -------------------------------------------------------------- feed_pages

2
tests/test_threaded_pipeline.py
View File

@ -18,7 +18,7 @@ from docling.pipeline.threaded_standard_pdf_pipeline import ThreadedStandardPdfP
def test_threaded_pipeline_multiple_documents(): def test_threaded_pipeline_multiple_documents():
"""Test threaded pipeline with multiple documents and compare with standard pipeline""" """Test threaded pipeline with multiple documents and compare with standard pipeline"""
test_files = [ test_files = [str(f) for f in Path("tests/data/pdf").rglob("*.pdf")] or [
"tests/data/pdf/2203.01017v2.pdf", "tests/data/pdf/2203.01017v2.pdf",
"tests/data/pdf/2206.01062.pdf", "tests/data/pdf/2206.01062.pdf",
"tests/data/pdf/2305.03393v1.pdf", "tests/data/pdf/2305.03393v1.pdf",