style: apply black formatting to type-annotated files

Author: scyyh11

paddleocr/_abstract.py

@@ -21,7 +21,9 @@
 
 class CLISubcommandExecutor(metaclass=abc.ABCMeta):
     @abc.abstractmethod
-    def add_subparser(self, subparsers: argparse._SubParsersAction) -> argparse.ArgumentParser:
+    def add_subparser(
+        self, subparsers: argparse._SubParsersAction
+    ) -> argparse.ArgumentParser:
         raise NotImplementedError
 
     @abc.abstractmethod

paddleocr/_common_args.py

@@ -33,7 +33,9 @@
 from ._utils.cli import str2bool
 
 
-def parse_common_args(kwargs: dict[str, Any], *, default_enable_hpi: bool | None) -> dict[str, Any]:
+def parse_common_args(
+    kwargs: dict[str, Any], *, default_enable_hpi: bool | None
+) -> dict[str, Any]:
     default_vals = {
         "device": DEFAULT_DEVICE,
         "enable_hpi": default_enable_hpi,
@@ -62,7 +64,9 @@ def parse_common_args(kwargs: dict[str, Any], *, default_enable_hpi: bool | None
     return kwargs
 
 
-def prepare_common_init_args(model_name: str | None, common_args: dict[str, Any]) -> dict[str, Any]:
+def prepare_common_init_args(
+    model_name: str | None, common_args: dict[str, Any]
+) -> dict[str, Any]:
     device = common_args["device"]
     if device is None:
         device = get_default_device()
@@ -99,7 +103,12 @@ def prepare_common_init_args(model_name: str | None, common_args: dict[str, Any]
     return init_kwargs
 
 
-def add_common_cli_opts(parser: argparse.ArgumentParser, *, default_enable_hpi: bool | None, allow_multiple_devices: bool) -> None:
+def add_common_cli_opts(
+    parser: argparse.ArgumentParser,
+    *,
+    default_enable_hpi: bool | None,
+    allow_multiple_devices: bool,
+) -> None:
     if allow_multiple_devices:
         help_ = "Device(s) to use for inference, e.g., `cpu`, `gpu`, `npu`, `gpu:0`, `gpu:0,1`. If multiple devices are specified, inference will be performed in parallel. Note that parallel inference is not always supported. By default, GPU 0 will be used if available; otherwise, the CPU will be used."
     else:

paddleocr/_models/base.py

@@ -93,7 +93,9 @@ class PredictorCLISubcommandExecutor(CLISubcommandExecutor):
     def subparser_name(self) -> str:
         raise NotImplementedError
 
-    def add_subparser(self, subparsers: argparse._SubParsersAction) -> argparse.ArgumentParser:
+    def add_subparser(
+        self, subparsers: argparse._SubParsersAction
+    ) -> argparse.ArgumentParser:
         subparser = subparsers.add_parser(name=self.subparser_name)
         self._update_subparser(subparser)
         subparser.add_argument("--model_name", type=str, help="Name of the model.")

paddleocr/_pipelines/base.py

@@ -119,7 +119,9 @@ class PipelineCLISubcommandExecutor(CLISubcommandExecutor):
     def subparser_name(self) -> str:
         raise NotImplementedError
 
-    def add_subparser(self, subparsers: argparse._SubParsersAction) -> argparse.ArgumentParser:
+    def add_subparser(
+        self, subparsers: argparse._SubParsersAction
+    ) -> argparse.ArgumentParser:
         subparser = subparsers.add_parser(name=self.subparser_name)
         self._update_subparser(subparser)
         add_common_cli_opts(

paddleocr/_pipelines/paddleocr_vl.py

@@ -222,7 +222,11 @@ def concatenate_markdown_pages(self, markdown_list: list[str]) -> str:
         return self.paddlex_pipeline.concatenate_markdown_pages(markdown_list)
 
     def restructure_pages(
-        self, res_list: list[Any], merge_tables: bool = True, relevel_titles: bool = True, concatenate_pages: bool = False
+        self,
+        res_list: list[Any],
+        merge_tables: bool = True,
+        relevel_titles: bool = True,
+        concatenate_pages: bool = False,
     ) -> list[Any]:
         return list(
             self.paddlex_pipeline.restructure_pages(

paddleocr/_pipelines/pp_chatocrv4_doc.py

@@ -88,14 +88,21 @@ def __init__(
     def _paddlex_pipeline_name(self) -> str:
         return "PP-ChatOCRv4-doc"
 
-    def save_vector(self, vector_info: Any, save_path: str, retriever_config: dict[str, Any] | None = None) -> Any:
+    def save_vector(
+        self,
+        vector_info: Any,
+        save_path: str,
+        retriever_config: dict[str, Any] | None = None,
+    ) -> Any:
         return self.paddlex_pipeline.save_vector(
             vector_info=vector_info,
             save_path=save_path,
             retriever_config=retriever_config,
         )
 
-    def load_vector(self, data_path: str, retriever_config: dict[str, Any] | None = None) -> Any:
+    def load_vector(
+        self, data_path: str, retriever_config: dict[str, Any] | None = None
+    ) -> Any:
         return self.paddlex_pipeline.load_vector(
             data_path=data_path, retriever_config=retriever_config
         )
@@ -233,7 +240,13 @@ def build_vector(
             retriever_config=retriever_config,
         )
 
-    def mllm_pred(self, input: InputType, key_list: list[str], *, mllm_chat_bot_config: dict[str, Any] | None = None) -> Any:
+    def mllm_pred(
+        self,
+        input: InputType,
+        key_list: list[str],
+        *,
+        mllm_chat_bot_config: dict[str, Any] | None = None,
+    ) -> Any:
         return self.paddlex_pipeline.mllm_pred(
             input,
             key_list,

paddleocr/_pipelines/pp_structurev3.py

@@ -534,7 +534,9 @@ def _get_paddlex_config_overrides(self) -> dict[str, Any]:
         }
         return create_config_from_structure(STRUCTURE)
 
-    def _get_ocr_model_names(self, lang: str | None, ppocr_version: str | None) -> tuple[str | None, str | None]:
+    def _get_ocr_model_names(
+        self, lang: str | None, ppocr_version: str | None
+    ) -> tuple[str | None, str | None]:
         LATIN_LANGS = [
             "af",
             "az",

paddleocr/_pipelines/utils.py

@@ -17,7 +17,12 @@
 from typing import Any
 
 
-def create_config_from_structure(structure: dict[str, Any], *, unset: Any = None, config: dict[str, Any] | None = None) -> dict[str, Any]:
+def create_config_from_structure(
+    structure: dict[str, Any],
+    *,
+    unset: Any = None,
+    config: dict[str, Any] | None = None,
+) -> dict[str, Any]:
     if config is None:
         config = {}
     for k, v in structure.items():

paddleocr/_utils/cli.py

@@ -32,7 +32,9 @@ def get_subcommand_args(args: argparse.Namespace) -> dict[str, Any]:
     return args_dict
 
 
-def add_simple_inference_args(subparser: argparse.ArgumentParser, *, input_help: str | None = None) -> None:
+def add_simple_inference_args(
+    subparser: argparse.ArgumentParser, *, input_help: str | None = None
+) -> None:
     if input_help is None:
         input_help = "Input path or URL."
     subparser.add_argument(
@@ -49,7 +51,11 @@ def add_simple_inference_args(subparser: argparse.ArgumentParser, *, input_help:
     )
 
 
-def perform_simple_inference(wrapper_cls: type, params: dict[str, Any], predict_param_names: set[str] | None = None) -> None:
+def perform_simple_inference(
+    wrapper_cls: type,
+    params: dict[str, Any],
+    predict_param_names: set[str] | None = None,
+) -> None:
     params = params.copy()
 
     input_ = params.pop("input")