paddlex 3.0.0b2__py3-none-any.whl → 3.0.0rc1__py3-none-any.whl

paddlex/repo_apis/PaddleOCR_api/text_rec/register.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,12 +14,11 @@
 
 import os
 import os.path as osp
-from pathlib import Path
 
 from ...base.register import register_model_info, register_suite_info
+from .config import TextRecConfig
 from .model import TextRecModel
 from .runner import TextRecRunner
-from .config import TextRecConfig
 
 REPO_ROOT_PATH = os.environ.get("PADDLE_PDX_PADDLEOCR_PATH")
 PDX_CONFIG_DIR = osp.abspath(osp.join(osp.dirname(__file__), "..", "configs"))
@@ -34,6 +33,116 @@ register_suite_info(
     }
 )
 
+register_model_info(
+    {
+        "model_name": "PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "en_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "en_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+
+register_model_info(
+    {
+        "model_name": "korean_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "korean_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "japan_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "japan_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "chinese_cht_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "chinese_cht_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "te_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "te_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "ka_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "ka_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "ta_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "ta_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "latin_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "latin_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "arabic_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "arabic_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "cyrillic_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "cyrillic_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "devanagari_PP-OCRv3_mobile_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "devanagari_PP-OCRv3_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+
 register_model_info(
     {
         "model_name": "PP-OCRv4_mobile_rec",
@@ -54,27 +163,36 @@ register_model_info(
 
 register_model_info(
     {
-        "model_name": "ch_SVTRv2_rec",
+        "model_name": "en_PP-OCRv4_mobile_rec",
         "suite": "TextRec",
-        "config_path": osp.join(PDX_CONFIG_DIR, "ch_SVTRv2_rec.yaml"),
-        "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
+        "config_path": osp.join(PDX_CONFIG_DIR, "en_PP-OCRv4_mobile_rec.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
     }
 )
 
 register_model_info(
     {
-        "model_name": "ch_RepSVTR_rec",
+        "model_name": "PP-OCRv4_server_rec_doc",
         "suite": "TextRec",
-        "config_path": osp.join(PDX_CONFIG_DIR, "ch_RepSVTR_rec.yaml"),
+        "config_path": osp.join(PDX_CONFIG_DIR, "PP-OCRv4_server_rec_doc.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+register_model_info(
+    {
+        "model_name": "ch_SVTRv2_rec",
+        "suite": "TextRec",
+        "config_path": osp.join(PDX_CONFIG_DIR, "ch_SVTRv2_rec.yaml"),
         "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
     }
 )
 
 register_model_info(
     {
-        "model_name": "LaTeX_OCR_rec",
+        "model_name": "ch_RepSVTR_rec",
         "suite": "TextRec",
-        "config_path": osp.join(PDX_CONFIG_DIR, "LaTeX_OCR_rec.yml"),
+        "config_path": osp.join(PDX_CONFIG_DIR, "ch_RepSVTR_rec.yaml"),
         "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
     }
 )

paddlex/repo_apis/PaddleOCR_api/text_rec/runner.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,7 +14,6 @@
 
 
 import os
-import tempfile
 
 from ...base import BaseRunner
 from ...base.utils.subprocess import CompletedProcess

paddlex/repo_apis/PaddleSeg_api/__init__.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

paddlex/repo_apis/PaddleSeg_api/base_seg_config.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,8 +16,8 @@ from urllib.parse import urlparse
 
 import yaml
 
-from ..base import BaseConfig
 from ...utils.misc import abspath
+from ..base import BaseConfig
 
 
 class BaseSegConfig(BaseConfig):

paddlex/repo_apis/PaddleSeg_api/seg/__init__.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

paddlex/repo_apis/PaddleSeg_api/seg/config.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,14 +14,11 @@
 
 
 import os
-from functools import lru_cache
-
-import yaml
 from typing import Union
 
-from ..base_seg_config import BaseSegConfig
-from ....utils.misc import abspath
 from ....utils import logging
+from ....utils.misc import abspath
+from ..base_seg_config import BaseSegConfig
 
 
 class SegConfig(BaseSegConfig):
@@ -72,6 +69,15 @@ class SegConfig(BaseSegConfig):
         if "model" in self:
             self.model["num_classes"] = num_classes
 
+        if self.model_name in ["MaskFormer_tiny", "MaskFormer_small"]:
+            for tf_cfg in self.train_dataset["transforms"]:
+                if tf_cfg["type"] == "GenerateInstanceTargets":
+                    tf_cfg["num_classes"] = num_classes
+
+            losses = self.loss["types"]
+            for loss_cfg in losses:
+                loss_cfg["num_classes"] = num_classes
+
     def update_train_crop_size(self, crop_size: Union[int, list]):
         """update the image cropping size of training preprocessing
 

paddlex/repo_apis/PaddleSeg_api/seg/model.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,13 +14,13 @@
 
 import os
 
+from ....utils.cache import DEFAULT_CACHE_DIR
+from ....utils.device import parse_device
+from ....utils.download import download
+from ....utils.misc import abspath
 from ...base import BaseModel
 from ...base.utils.arg import CLIArgument
 from ...base.utils.subprocess import CompletedProcess
-from ....utils.device import parse_device
-from ....utils.misc import abspath
-from ....utils.download import download
-from ....utils.cache import DEFAULT_CACHE_DIR
 
 
 class SegModel(BaseModel):
@@ -126,6 +126,10 @@ class SegModel(BaseModel):
         if log_iters is not None:
             cli_args.append(CLIArgument("--log_iters", log_iters))
 
+        input_shape = kwargs.pop("input_shape", None)
+        if input_shape is not None:
+            cli_args.append(CLIArgument("--input_shape", *input_shape))
+
         # Benchmarking mode settings
         benchmark = kwargs.pop("benchmark", None)
         if benchmark is not None:
@@ -167,8 +171,11 @@ class SegModel(BaseModel):
 
         # PDX related settings
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
+        export_with_pir = kwargs.pop("export_with_pir", False)
         config.set_val("uniform_output_enabled", uniform_output_enabled)
         config.set_val("pdx_model_name", self.name)
+        if export_with_pir:
+            config.set_val("export_with_pir", export_with_pir)
 
         self._assert_empty_kwargs(kwargs)
 
@@ -352,8 +359,11 @@ class SegModel(BaseModel):
 
         # PDX related settings
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
+        export_with_pir = kwargs.pop("export_with_pir", False)
         config.set_val("uniform_output_enabled", uniform_output_enabled)
         config.set_val("pdx_model_name", self.name)
+        if export_with_pir:
+            config.set_val("export_with_pir", export_with_pir)
 
         self._assert_empty_kwargs(kwargs)
 

paddlex/repo_apis/PaddleSeg_api/seg/register.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,12 +14,11 @@
 
 import os
 import os.path as osp
-from pathlib import Path
 
 from ...base.register import register_model_info, register_suite_info
+from .config import SegConfig
 from .model import SegModel
 from .runner import SegRunner
-from .config import SegConfig
 
 REPO_ROOT_PATH = os.environ.get("PADDLE_PDX_PADDLESEG_PATH")
 PDX_CONFIG_DIR = osp.abspath(osp.join(osp.dirname(__file__), "..", "configs"))
@@ -238,6 +237,26 @@ register_model_info(
 )
 
 
+register_model_info(
+    {
+        "model_name": "MaskFormer_tiny",
+        "suite": "Seg",
+        "config_path": osp.join(PDX_CONFIG_DIR, "MaskFormer_tiny.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+
+register_model_info(
+    {
+        "model_name": "MaskFormer_small",
+        "suite": "Seg",
+        "config_path": osp.join(PDX_CONFIG_DIR, "MaskFormer_small.yaml"),
+        "supported_apis": ["train", "evaluate", "predict", "export"],
+    }
+)
+
+
 # For compatibility
 def _set_alias(model_name, alias):
     from ...base.register import get_registered_model_info

paddlex/repo_apis/PaddleSeg_api/seg/runner.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,7 +14,6 @@
 
 
 import os
-import tempfile
 
 from ...base import BaseRunner
 from ...base.utils.arg import gather_opts_args

paddlex/repo_apis/PaddleTS_api/__init__.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -13,7 +13,8 @@
 # limitations under the License.
 
 
+from .ts_ad import register
+from .ts_cls import register
+
 # Register models and suites
 from .ts_fc import register
-from .ts_cls import register
-from .ts_ad import register

paddlex/repo_apis/PaddleTS_api/ts_ad/__init__.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

paddlex/repo_apis/PaddleTS_api/ts_ad/config.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,10 +14,9 @@
 
 
 import os
-import yaml
 
-from ..ts_base.config import BaseTSConfig
 from ....utils.misc import abspath
+from ..ts_base.config import BaseTSConfig
 
 
 class TSAnomalyConfig(BaseTSConfig):

paddlex/repo_apis/PaddleTS_api/ts_ad/register.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -18,8 +18,8 @@ import os.path as osp
 
 from ...base.register import register_model_info, register_suite_info
 from ..ts_base.model import TSModel
-from .runner import TSADRunner
 from .config import TSAnomalyConfig
+from .runner import TSADRunner
 
 REPO_ROOT_PATH = os.environ.get("PADDLE_PDX_PADDLETS_PATH")
 PDX_CONFIG_DIR = osp.abspath(osp.join(osp.dirname(__file__), "..", "configs"))

paddlex/repo_apis/PaddleTS_api/ts_ad/runner.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -13,10 +13,10 @@
 # limitations under the License.
 
 
+from ....utils.errors import raise_unsupported_api_error
 from ...base import BaseRunner
 from ...base.utils.arg import gather_opts_args
 from ...base.utils.subprocess import CompletedProcess
-from ....utils.errors import raise_unsupported_api_error
 
 
 class TSADRunner(BaseRunner):

paddlex/repo_apis/PaddleTS_api/ts_base/__init__.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

paddlex/repo_apis/PaddleTS_api/ts_base/config.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -17,8 +17,8 @@ from urllib.parse import urlparse
 
 import ruamel.yaml
 
-from ...base import BaseConfig
 from ....utils.misc import abspath, convert_and_remove_types
+from ...base import BaseConfig
 
 
 class BaseTSConfig(BaseConfig):
@@ -57,6 +57,9 @@ class BaseTSConfig(BaseConfig):
         Args:
             config_file_path (str): the path to save self as yaml file.
         """
+        output_dir = os.path.dirname(config_file_path)
+        if not os.path.exists(output_dir):
+            os.makedirs(output_dir)
         yaml = ruamel.yaml.YAML()
         with open(config_file_path, "w", encoding="utf-8") as f:
             dict_to_dump = self.dict
@@ -71,6 +74,26 @@ class BaseTSConfig(BaseConfig):
         """
         self.update({"epoch": epochs})
 
+    def update_to_static(self, dy2st: bool):
+        """update config to set dynamic to static mode
+
+        Args:
+            dy2st (bool): whether or not to use the dynamic to static mode.
+        """
+        self.update({"to_static_train": dy2st})
+
+    def update_amp(self, amp: str = "O2"):
+        """update AMP settings
+
+        Args:
+            amp (None | str): the AMP level if it is not None or `OFF`.
+        """
+        _cfg = {
+            "use_amp": True if amp is not None else False,
+            "amp_level": amp,
+        }
+        self.update(_cfg)
+
     def update_weights(self, weight_path: str):
         """update weight path
 
@@ -146,7 +169,6 @@ class BaseTSConfig(BaseConfig):
             device (str): the running device to set
         """
         # PaddleTS does not support multi-device training currently.
-        pass
 
     def update_print_mem_info(self, print_mem_info: bool):
         """setting print memory info"""

paddlex/repo_apis/PaddleTS_api/ts_base/model.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,12 +14,12 @@
 
 import os
 
+from ....utils.device import parse_device
+from ....utils.errors import raise_unsupported_api_error
+from ....utils.misc import abspath
 from ...base import BaseModel
 from ...base.utils.arg import CLIArgument
 from ...base.utils.subprocess import CompletedProcess
-from ....utils.device import parse_device
-from ....utils.misc import abspath
-from ....utils.errors import raise_unsupported_api_error
 
 
 class TSModel(BaseModel):
@@ -73,11 +73,11 @@ class TSModel(BaseModel):
         if resume_path:
             raise ValueError("`resume_path` is not supported.")
         # No need to handle `ips`
-        if amp is not None and amp != "OFF":
-            raise ValueError(f"`amp`={amp} is not supported.")
-
-        if dy2st:
-            raise ValueError(f"`dy2st`={dy2st} is not supported.")
+        benchmark = kwargs.pop("benchmark", None)
+        if benchmark is not None:
+            amp = benchmark.get("amp", None)
+            if amp in ["O1", "O2"]:
+                config.update_amp(amp)
 
         if use_vdl:
             raise ValueError(f"`use_vdl`={use_vdl} is not supported.")
@@ -85,7 +85,6 @@ class TSModel(BaseModel):
         if device is not None:
             device_type, _ = parse_device(device)
             cli_args.append(CLIArgument("--device", device_type))
-
         if save_dir is not None:
             save_dir = abspath(save_dir)
         else:
@@ -111,8 +110,11 @@ class TSModel(BaseModel):
                 cli_args.append(CLIArgument("--num_workers", num_workers))
         # PDX related settings
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
+        export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"uniform_output_enabled": uniform_output_enabled})
         config.update({"pdx_model_name": self.name})
+        if export_with_pir:
+            config.update({"export_with_pir": export_with_pir})
 
         self._assert_empty_kwargs(kwargs)
 
@@ -235,13 +237,15 @@ class TSModel(BaseModel):
         if device is not None:
             device_type, _ = parse_device(device)
             cli_args.append(CLIArgument("--device", device_type))
-
+        export_with_pir = kwargs.pop("export_with_pir", False)
         self._assert_empty_kwargs(kwargs)
         with self._create_new_config_file() as config_path:
             # Update YAML config file
             config = self.config.copy()
             config.update_pretrained_weights(weight_path)
             config.update({"pdx_model_name": self.name})
+            if export_with_pir:
+                config.update({"export_with_pir": export_with_pir})
             config.dump(config_path)
 
             return self.runner.export(config_path, cli_args, device)

paddlex/repo_apis/PaddleTS_api/ts_base/runner.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -13,10 +13,10 @@
 # limitations under the License.
 
 
+from ....utils.errors import raise_unsupported_api_error
 from ...base import BaseRunner
 from ...base.utils.arg import gather_opts_args
 from ...base.utils.subprocess import CompletedProcess
-from ....utils.errors import raise_unsupported_api_error
 
 
 class TSRunner(BaseRunner):

paddlex/repo_apis/PaddleTS_api/ts_cls/__init__.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

paddlex/repo_apis/PaddleTS_api/ts_cls/config.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,10 +14,9 @@
 
 
 import os
-import yaml
 
-from ..ts_base.config import BaseTSConfig
 from ....utils.misc import abspath
+from ..ts_base.config import BaseTSConfig
 
 
 class TSClassifyConfig(BaseTSConfig):

paddlex/repo_apis/PaddleTS_api/ts_cls/register.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -18,8 +18,8 @@ import os.path as osp
 
 from ...base.register import register_model_info, register_suite_info
 from ..ts_base.model import TSModel
-from .runner import TSCLSRunner
 from .config import TSClassifyConfig
+from .runner import TSCLSRunner
 
 REPO_ROOT_PATH = os.environ.get("PADDLE_PDX_PADDLETS_PATH")
 PDX_CONFIG_DIR = osp.abspath(osp.join(osp.dirname(__file__), "..", "configs"))