paddlex 3.0.0rc0__py3-none-any.whl → 3.0.1__py3-none-any.whl

paddlex/modules/table_recognition/dataset_checker/dataset_src/check_dataset.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,12 +13,14 @@
 # limitations under the License.
 
 
-import os
 import json
+import os
 import os.path as osp
-from PIL import Image, ImageOps
 from collections import defaultdict
-from .....utils.errors import DatasetFileNotFoundError, CheckFailedError
+
+from PIL import Image, ImageOps
+
+from .....utils.errors import DatasetFileNotFoundError
 
 
 def check(dataset_dir, output, dataset_type="PubTabTableRecDataset", sample_num=10):
@@ -32,7 +34,6 @@ def check(dataset_dir, output, dataset_type="PubTabTableRecDataset", sample_num=
             raise DatasetFileNotFoundError(file_path=dataset_dir)
 
         tags = ["train", "val"]
-        max_recorded_sample_cnts = 50
         sample_cnts = dict()
         sample_paths = defaultdict(list)
 
@@ -55,8 +56,8 @@ def check(dataset_dir, output, dataset_type="PubTabTableRecDataset", sample_num=
                     for line in all_lines:
                         info = json.loads(line.strip("\n"))
                         file_name = info["filename"]
-                        cells = info["html"]["cells"].copy()
-                        structure = info["html"]["structure"]["tokens"].copy()
+                        info["html"]["cells"].copy()
+                        info["html"]["structure"]["tokens"].copy()
 
                         img_path = osp.join(dataset_dir, file_name)
 

paddlex/modules/table_recognition/dataset_checker/dataset_src/split_dataset.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.
@@ -15,6 +15,7 @@
 
 import os
 from random import shuffle
+
 from .....utils.file_interface import custom_open
 
 

paddlex/modules/table_recognition/evaluator.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.
@@ -23,7 +23,7 @@ class TableRecEvaluator(BaseEvaluator):
     entities = MODELS
 
     def update_config(self):
-        """update evalution config"""
+        """update evaluation config"""
         if self.eval_config.log_interval:
             self.pdx_config.update_log_interval(self.eval_config.log_interval)
 
@@ -32,7 +32,7 @@ class TableRecEvaluator(BaseEvaluator):
         )
 
     def get_eval_kwargs(self) -> dict:
-        """get key-value arguments of model evalution function
+        """get key-value arguments of model evaluation function
 
         Returns:
             dict: the arguments of evaluation function.

paddlex/modules/table_recognition/exportor.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/modules/table_recognition/model_list.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/modules/table_recognition/trainer.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,11 +13,7 @@
 # limitations under the License.
 
 
-import os
-from pathlib import Path
-
 from ..base import BaseTrainer
-from ...utils.config import AttrDict
 from .model_list import MODELS
 
 
@@ -67,4 +63,5 @@ class TableRecTrainer(BaseTrainer):
         return {
             "device": self.get_device(),
             "dy2st": self.train_config.get("dy2st", False),
+            "amp": self.train_config.get("amp", "OFF"),  # amp support 'O1', 'O2', 'OFF'
         }

paddlex/modules/text_detection/__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,6 +13,6 @@
 # limitations under the License.
 
 from .dataset_checker import TextDetDatasetChecker
-from .trainer import TextDetTrainer
 from .evaluator import TextDetEvaluator
 from .exportor import TextDetExportor
+from .trainer import TextDetTrainer

paddlex/modules/text_detection/dataset_checker/__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,17 +13,15 @@
 # limitations under the License.
 
 
+import json
 import os
 import os.path as osp
+from collections import Counter, defaultdict
 from pathlib import Path
-from collections import defaultdict, Counter
-
-import json
 
 from ...base import BaseDatasetChecker
-from .dataset_src import check, split_dataset, deep_analyse
-
 from ..model_list import MODELS
+from .dataset_src import check, deep_analyse, split_dataset
 
 
 class TextDetDatasetChecker(BaseDatasetChecker):

paddlex/modules/text_detection/dataset_checker/dataset_src/__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,6 +13,6 @@
 # limitations under the License.
 
 
+from .analyse_dataset import deep_analyse
 from .check_dataset import check
 from .split_dataset import split_dataset
-from .analyse_dataset import deep_analyse

paddlex/modules/text_detection/dataset_checker/dataset_src/analyse_dataset.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,20 +13,22 @@
 # limitations under the License.
 
 
-from cProfile import label
+import json
 import os
-
 from collections import defaultdict
 
-import matplotlib.pyplot as plt
-from matplotlib.backends.backend_agg import FigureCanvasAgg
 import numpy as np
 from PIL import Image, ImageOps
-import cv2
-import json
 
+from .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.file_interface import custom_open
 
+if is_dep_available("opencv-contrib-python"):
+    import cv2
+if is_dep_available("matplotlib"):
+    import matplotlib.pyplot as plt
+    from matplotlib.backends.backend_agg import FigureCanvasAgg
+
 
 # show data samples
 def simple_analyse(dataset_path, max_recorded_sample_cnts=20, show_label=True):
@@ -49,7 +51,6 @@ def simple_analyse(dataset_path, max_recorded_sample_cnts=20, show_label=True):
     lab_infos = defaultdict(list)
     res = [None] * 9
     delim = "\t"
-    valid_num_parts = 2
 
     for tag in tags:
         file_list = os.path.join(dataset_path, f"{tag}.txt")
@@ -142,6 +143,7 @@ def simple_analyse(dataset_path, max_recorded_sample_cnts=20, show_label=True):
         )
 
 
+@function_requires_deps("opencv-contrib-python")
 def show_label_img(img_path, dt_boxes):
     """draw ocr detection label"""
     img = cv2.imread(img_path)
@@ -151,13 +153,14 @@ def show_label_img(img_path, dt_boxes):
     return img[:, :, ::-1]
 
 
+@function_requires_deps("matplotlib", "opencv-contrib-python")
 def deep_analyse(dataset_path, output):
     """class analysis for dataset"""
     sample_results = simple_analyse(
         dataset_path, max_recorded_sample_cnts=float("inf"), show_label=False
     )
     lab_infos = sample_results[-3] + sample_results[-2] + sample_results[-1]
-    labels_cnt = defaultdict(int)
+    defaultdict(int)
     img_shapes = []  # w, h
     ratios_w = []
     ratios_h = []

paddlex/modules/text_detection/dataset_checker/dataset_src/check_dataset.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,13 +13,13 @@
 # limitations under the License.
 
 
+import json
 import os
 import os.path as osp
 from collections import defaultdict
 
-from PIL import Image, ImageOps
-import json
 import numpy as np
+from PIL import Image, ImageOps
 
 from .....utils.errors import DatasetFileNotFoundError
 

paddlex/modules/text_detection/dataset_checker/dataset_src/split_dataset.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.
@@ -15,9 +15,10 @@
 
 import glob
 import os.path
-import numpy as np
 import shutil
 
+import numpy as np
+
 from .....utils.file_interface import custom_open
 from .....utils.logging import info
 
@@ -71,7 +72,6 @@ def split_dataset(dataset_root, train_rate, val_rate):
     start = 0
     rate_list = [train_rate, val_rate]
     name_list = ["train", "val"]
-    separator = " "
     for i, name in enumerate(name_list):
         info("Creating {}.txt...".format(name))
 

paddlex/modules/text_detection/evaluator.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.
@@ -23,14 +23,14 @@ class TextDetEvaluator(BaseEvaluator):
     entities = MODELS
 
     def update_config(self):
-        """update evalution config"""
+        """update evaluation config"""
         if self.eval_config.log_interval:
             self.pdx_config.update_log_interval(self.eval_config.log_interval)
 
         self.pdx_config.update_dataset(self.global_config.dataset_dir, "TextDetDataset")
 
     def get_eval_kwargs(self) -> dict:
-        """get key-value arguments of model evalution function
+        """get key-value arguments of model evaluation function
 
         Returns:
             dict: the arguments of evaluation function.

paddlex/modules/text_detection/exportor.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/modules/text_detection/model_list.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,6 +13,8 @@
 # limitations under the License.
 
 MODELS = [
+    "PP-OCRv5_mobile_det",
+    "PP-OCRv5_server_det",
     "PP-OCRv4_mobile_det",
     "PP-OCRv4_server_det",
     "PP-OCRv4_mobile_seal_det",

paddlex/modules/text_detection/trainer.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,11 +13,7 @@
 # limitations under the License.
 
 
-import os
-from pathlib import Path
-
 from ..base import BaseTrainer
-from ...utils.config import AttrDict
 from .model_list import MODELS
 
 
@@ -65,4 +61,5 @@ class TextDetTrainer(BaseTrainer):
         return {
             "device": self.get_device(),
             "dy2st": self.train_config.get("dy2st", False),
+            "amp": self.train_config.get("amp", "OFF"),  # amp support 'O1', 'O2', 'OFF'
         }

paddlex/modules/text_recognition/__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,6 +13,6 @@
 # limitations under the License.
 
 from .dataset_checker import TextRecDatasetChecker
-from .trainer import TextRecTrainer
 from .evaluator import TextRecEvaluator
 from .exportor import TextRecExportor
+from .trainer import TextRecTrainer

paddlex/modules/text_recognition/dataset_checker/__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,18 +13,17 @@
 # limitations under the License.
 
 
+import json
 import os
 import os.path as osp
+from collections import Counter, defaultdict
 from pathlib import Path
-from collections import defaultdict, Counter
 
 from PIL import Image
-import json
 
 from ...base import BaseDatasetChecker
-from .dataset_src import check, split_dataset, deep_analyse, convert
-
 from ..model_list import MODELS
+from .dataset_src import check, convert, deep_analyse, split_dataset
 
 
 class TextRecDatasetChecker(BaseDatasetChecker):

paddlex/modules/text_recognition/dataset_checker/dataset_src/__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,7 @@
 # limitations under the License.
 
 
+from .analyse_dataset import deep_analyse
 from .check_dataset import check
 from .convert_dataset import convert
 from .split_dataset import split_dataset
-from .analyse_dataset import deep_analyse

paddlex/modules/text_recognition/dataset_checker/dataset_src/analyse_dataset.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,23 +13,24 @@
 # limitations under the License.
 
 
-import os
-import json
 import math
+import os
 import platform
-from pathlib import Path
-
 from collections import defaultdict
-from PIL import Image
-import cv2
+
 import numpy as np
-import matplotlib.pyplot as plt
-from matplotlib.backends.backend_agg import FigureCanvasAgg
-from matplotlib import font_manager
 
+from .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.file_interface import custom_open
-from .....utils.logging import warning
 from .....utils.fonts import PINGFANG_FONT_FILE_PATH
+from .....utils.logging import warning
+
+if is_dep_available("opencv-contrib-python"):
+    import cv2
+if is_dep_available("matplotlib"):
+    import matplotlib.pyplot as plt
+    from matplotlib import font_manager
+    from matplotlib.backends.backend_agg import FigureCanvasAgg
 
 
 def simple_analyse(dataset_path, images_dict):
@@ -77,10 +78,10 @@ def simple_analyse(dataset_path, images_dict):
     )
 
 
+@function_requires_deps("matplotlib", "opencv-contrib-python")
 def deep_analyse(dataset_path, output, datatype="MSTextRecDataset"):
     """class analysis for dataset"""
     tags = ["train", "val"]
-    all_instances = 0
     labels_cnt = {}
     x_max = []
     classes_max = []

paddlex/modules/text_recognition/dataset_checker/dataset_src/check_dataset.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,10 +18,8 @@ import os.path as osp
 from collections import defaultdict
 
 from PIL import Image, ImageOps
-import json
-import numpy as np
 
-from .....utils.errors import DatasetFileNotFoundError, CheckFailedError
+from .....utils.errors import CheckFailedError, DatasetFileNotFoundError
 
 
 def check(
@@ -35,7 +33,6 @@ def check(
         tags = ["train", "val"]
         delim = "\t"
         valid_num_parts = 2
-        max_recorded_sample_cnts = 50
         sample_cnts = dict()
         sample_paths = defaultdict(list)
         if dataset_type == "LaTeXOCRDataset":

paddlex/modules/text_recognition/dataset_checker/dataset_src/convert_dataset.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,23 +13,24 @@
 # limitations under the License.
 
 
-import os
-import shutil
-import json
-import random
 import math
+import os
 import pickle
-from tqdm import tqdm
 from collections import defaultdict
-import imagesize
+
+from .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.errors import ConvertFailedError
-from .....utils.logging import info, warning
+
+if is_dep_available("imagesize"):
+    import imagesize
+if is_dep_available("tqdm"):
+    from tqdm import tqdm
 
 
 def check_src_dataset(root_dir, dataset_type):
     """check src dataset format validity"""
     if dataset_type in ("MSTextRecDataset"):
-        anno_suffix = ".txt"
+        pass
     else:
         raise ConvertFailedError(
             message=f"数据格式转换失败！不支持{dataset_type}格式数据集。当前仅支持 MSTextRecDataset 格式。"
@@ -65,12 +66,12 @@ def convert_pkl_dataset(root_dir):
         txt2pickle(src_img_dir, src_anno_path, root_dir)
 
 
+@function_requires_deps("tqdm", "imagesize")
 def txt2pickle(images, equations, save_dir):
     phase = os.path.basename(equations).replace(".txt", "")
     save_p = os.path.join(save_dir, "latexocr_{}.pkl".format(phase))
     min_dimensions = (32, 32)
     max_dimensions = (672, 192)
-    max_length = 512
     data = defaultdict(lambda: [])
     pic_num = 0
     if images is not None and equations is not None:

paddlex/modules/text_recognition/dataset_checker/dataset_src/split_dataset.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 shutil
 from random import shuffle
 
 from .....utils.file_interface import custom_open

paddlex/modules/text_recognition/evaluator.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.
@@ -25,7 +25,7 @@ class TextRecEvaluator(BaseEvaluator):
     entities = MODELS
 
     def update_config(self):
-        """update evalution config"""
+        """update evaluation config"""
         if self.eval_config.log_interval:
             self.pdx_config.update_log_interval(self.eval_config.log_interval)
         if self.global_config["model"] == "LaTeX_OCR_rec":
@@ -53,7 +53,7 @@ class TextRecEvaluator(BaseEvaluator):
             self.pdx_config.update_label_dict_path(label_dict_path)
 
     def get_eval_kwargs(self) -> dict:
-        """get key-value arguments of model evalution function
+        """get key-value arguments of model evaluation function
 
         Returns:
             dict: the arguments of evaluation function.

paddlex/modules/text_recognition/exportor.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/modules/text_recognition/model_list.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.
@@ -31,4 +31,6 @@ MODELS = [
     "PP-OCRv4_server_rec_doc",
     "ch_SVTRv2_rec",
     "ch_RepSVTR_rec",
+    "PP-OCRv5_server_rec",
+    "PP-OCRv5_mobile_rec",
 ]

paddlex/modules/text_recognition/trainer.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,12 +13,10 @@
 # limitations under the License.
 
 
-import os
 import shutil
 from pathlib import Path
 
 from ..base import BaseTrainer
-from ...utils.config import AttrDict
 from .model_list import MODELS
 
 
@@ -103,4 +101,5 @@ class TextRecTrainer(BaseTrainer):
         return {
             "device": self.get_device(),
             "dy2st": self.train_config.get("dy2st", False),
+            "amp": self.train_config.get("amp", "OFF"),  # amp support 'O1', 'O2', 'OFF'
         }

paddlex/modules/ts_anomaly_detection/__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.
@@ -14,6 +14,6 @@
 
 
 from .dataset_checker import TSADDatasetChecker
-from .trainer import TSADTrainer
 from .evaluator import TSADEvaluator
 from .exportor import TSADExportor
+from .trainer import TSADTrainer