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

fix: Code styling

Browse Source 
Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>
This commit is contained in:
Nikos Livathinos 2024-12-08 22:14:48 +01:00
parent 64c7382880
commit 04977aac9f
3 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docling/datamodel/pipeline_options.py
View File

@ -143,7 +143,11 @@ class PdfPipelineOptions(PipelineOptions):
table_structure_options: TableStructureOptions = TableStructureOptions() table_structure_options: TableStructureOptions = TableStructureOptions()
ocr_options: Union[ ocr_options: Union[
EasyOcrOptions, TesseractCliOcrOptions, TesseractOcrOptions, OcrMacOptions, RapidOcrOptions EasyOcrOptions,
TesseractCliOcrOptions,
TesseractOcrOptions,
OcrMacOptions,
RapidOcrOptions,
] = Field(EasyOcrOptions(), discriminator="kind") ] = Field(EasyOcrOptions(), discriminator="kind")
images_scale: float = 1.0 images_scale: float = 1.0

4
docling/pipeline/standard_pdf_pipeline.py
View File

@ -96,11 +96,13 @@ class StandardPdfPipeline(PaginatedPipeline):
def download_models_hf( def download_models_hf(
local_dir: Optional[Path] = None, force: bool = False local_dir: Optional[Path] = None, force: bool = False
) -> Path: ) -> Path:
from functools import partialmethod
from huggingface_hub import snapshot_download from huggingface_hub import snapshot_download
# Disable tqdm prints used by HF # Disable tqdm prints used by HF
from tqdm import tqdm from tqdm import tqdm
from functools import partialmethod
tqdm.__init__ = partialmethod(tqdm.__init__, disable=True) tqdm.__init__ = partialmethod(tqdm.__init__, disable=True)
download_path = snapshot_download( download_path = snapshot_download(