From 4918e11cbe60b583d369804957e7f284f82c0ac0 Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Wed, 16 Oct 2024 19:42:02 +0200 Subject: [PATCH] apply pre-commit with new path Signed-off-by: Michele Dolfi --- .pre-commit-config.yaml | 8 ++++---- docs/examples/batch_convert.py | 2 +- docs/examples/custom_convert.py | 7 ++----- docs/examples/develop_picture_enrichment.py | 21 +++++++++++++++------ docs/examples/export_multimodal.py | 2 +- docs/examples/export_tables.py | 2 +- docs/examples/run_with_formats.py | 5 ++++- 7 files changed, 28 insertions(+), 19 deletions(-) diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 6ab5c1aa..19bb27c3 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -4,13 +4,13 @@ repos: hooks: - id: black name: Black - entry: poetry run black docling examples tests + entry: poetry run black docling docs/examples tests pass_filenames: false language: system files: '\.py$' - id: isort name: isort - entry: poetry run isort docling examples tests + entry: poetry run isort docling docs/examples tests pass_filenames: false language: system files: '\.py$' @@ -28,13 +28,13 @@ repos: files: '\.py$' - id: nbqa_black name: nbQA Black - entry: poetry run nbqa black examples + entry: poetry run nbqa black docs/examples pass_filenames: false language: system files: '\.ipynb$' - id: nbqa_isort name: nbQA isort - entry: poetry run nbqa isort examples + entry: poetry run nbqa isort docs/examples pass_filenames: false language: system files: '\.ipynb$' diff --git a/docs/examples/batch_convert.py b/docs/examples/batch_convert.py index c8e244fa..da1c701f 100644 --- a/docs/examples/batch_convert.py +++ b/docs/examples/batch_convert.py @@ -1,9 +1,9 @@ import json import logging +import time from pathlib import Path from typing import Iterable -import time import yaml from docling.datamodel.base_models import ConversionStatus diff --git a/docs/examples/custom_convert.py b/docs/examples/custom_convert.py index f6848aaa..249c7f78 100644 --- a/docs/examples/custom_convert.py +++ b/docs/examples/custom_convert.py @@ -1,13 +1,10 @@ import json import logging +import time from pathlib import Path -import time - from docling.datamodel.base_models import InputFormat -from docling.datamodel.pipeline_options import ( - PdfPipelineOptions, -) +from docling.datamodel.pipeline_options import PdfPipelineOptions from docling.document_converter import DocumentConverter, PdfFormatOption _log = logging.getLogger(__name__) diff --git a/docs/examples/develop_picture_enrichment.py b/docs/examples/develop_picture_enrichment.py index 64b99e49..9e2cfb8a 100644 --- a/docs/examples/develop_picture_enrichment.py +++ b/docs/examples/develop_picture_enrichment.py @@ -2,8 +2,13 @@ import logging from pathlib import Path from typing import Any, Iterable -from docling_core.types.doc import DoclingDocument, NodeItem -from docling_core.types.doc import PictureClassificationData, PictureItem, PictureClassificationClass +from docling_core.types.doc import ( + DoclingDocument, + NodeItem, + PictureClassificationClass, + PictureClassificationData, + PictureItem, +) from docling.datamodel.base_models import InputFormat from docling.datamodel.pipeline_options import PdfPipelineOptions @@ -36,10 +41,14 @@ class ExamplePictureClassifierEnrichmentModel(BaseEnrichmentModel): # uncomment this to interactively visualize the image # element.image.pil_image.show() - element.annotations.append(PictureClassificationData( - provenance="example_classifier-0.0.1", - predicted_classes=[PictureClassificationClass(class_name="dummy", confidence=0.42)] - )) + element.annotations.append( + PictureClassificationData( + provenance="example_classifier-0.0.1", + predicted_classes=[ + PictureClassificationClass(class_name="dummy", confidence=0.42) + ], + ) + ) yield element diff --git a/docs/examples/export_multimodal.py b/docs/examples/export_multimodal.py index 1de714ed..09885bd3 100644 --- a/docs/examples/export_multimodal.py +++ b/docs/examples/export_multimodal.py @@ -1,9 +1,9 @@ import datetime import logging +import time from pathlib import Path import pandas as pd -import time from docling.datamodel.base_models import InputFormat from docling.datamodel.pipeline_options import PdfPipelineOptions diff --git a/docs/examples/export_tables.py b/docs/examples/export_tables.py index 55d3a40f..68b9ce47 100644 --- a/docs/examples/export_tables.py +++ b/docs/examples/export_tables.py @@ -1,8 +1,8 @@ import logging +import time from pathlib import Path import pandas as pd -import time from docling.document_converter import DocumentConverter diff --git a/docs/examples/run_with_formats.py b/docs/examples/run_with_formats.py index eeeee66c..758d7cc6 100644 --- a/docs/examples/run_with_formats.py +++ b/docs/examples/run_with_formats.py @@ -16,6 +16,7 @@ from docling.pipeline.standard_pdf_pipeline import StandardPdfPipeline _log = logging.getLogger(__name__) + def main(): input_paths = [ Path("tests/data/wiki_duck.html"), @@ -69,5 +70,7 @@ def main(): with (out_path / f"{res.input.file.name}.yaml").open("w") as fp: fp.write(yaml.safe_dump(res.document.export_to_dict())) + + if __name__ == "__main__": - main() \ No newline at end of file + main()