修改tk名字为mindpet

.github/workflows/ci.yml

@@ -31,7 +31,7 @@ jobs:
         pip install -r requirements/ci_requirements.txt
     - name: Analysing the code with pylint
       run: |
-        pylint tk --rcfile=.github/pylint.conf
+        pylint mindpet --rcfile=.github/pylint.conf
     - name: Test with unit test (UT) pytest
       run: |
         pytest test/unit_test

tk/__init__.py → mindpet/__init__.py

@@ -2,6 +2,6 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-import tk.tk_sdk as tk_sdk
+import mindpet.tk_sdk as tk_sdk
 
 __all__ = ["tk_sdk"]

mindpet/delta/__init__.py

@@ -0,0 +1,12 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
+
+from mindpet.delta.lora import LoRADense
+from mindpet.delta.prefix_layer import PrefixLayer
+from mindpet.delta.low_rank_adapter import LowRankAdapterDense, LowRankAdapterLayer
+from mindpet.delta.adapter import AdapterDense, AdapterLayer
+from mindpet.delta.r_drop import RDropLoss, rdrop_repeat
+
+__all__ = ['LoRADense', 'PrefixLayer', 'LowRankAdapterDense', 'LowRankAdapterLayer',
+           'AdapterDense', 'AdapterLayer', 'RDropLoss', 'rdrop_repeat']

tk/delta/adapter.py → mindpet/delta/adapter.py

@@ -10,7 +10,7 @@
 from mindspore.nn.layer.activation import get_activation, _activation
 from mindspore.ops import operations as P
 from mindspore.ops import functional as F
-from tk.delta.delta_constants import VALID_TENSOR_DATATYPE
+from mindpet.delta.delta_constants import VALID_TENSOR_DATATYPE
 
 try:
     from mindspore.nn.transformer.layers import _Linear, _args_type_validator_check, _valid_value_checks

tk/delta/lora.py → mindpet/delta/lora.py

@@ -12,8 +12,8 @@
 from mindspore.ops import operations as P
 from mindspore.ops import functional as F
 from mindspore.common.initializer import initializer, HeUniform
-from tk.delta.delta_constants import VALID_TENSOR_DATATYPE
-from tk.utils.version_control import get_dropout
+from mindpet.delta.delta_constants import VALID_TENSOR_DATATYPE
+from mindpet.utils.version_control import get_dropout
 
 try:
     from mindspore._checkparam import Validator, Rel

tk/delta/low_rank_adapter.py → mindpet/delta/low_rank_adapter.py

@@ -14,7 +14,7 @@
 from mindspore.nn.layer.activation import get_activation, _activation
 from mindspore.ops import operations as P
 from mindspore.ops import functional as F
-from tk.delta.delta_constants import VALID_TENSOR_DATATYPE
+from mindpet.delta.delta_constants import VALID_TENSOR_DATATYPE
 
 try:
     from mindspore.nn.transformer.layers import _args_type_validator_check, _valid_type_checks, _valid_value_checks

tk/delta/prefix_layer.py → mindpet/delta/prefix_layer.py

@@ -5,7 +5,7 @@
 import mindspore as ms
 import mindspore.nn as nn
 
-from tk.utils.version_control import get_dropout
+from mindpet.utils.version_control import get_dropout
 
 try:
     from mindspore._checkparam import Validator, Rel

tk/graph/__init__.py → mindpet/graph/__init__.py

@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-from tk.graph.freeze_utils import freeze_modules, freeze_delta, freeze_from_config
-from tk.graph.ckpt_util import TrainableParamsCheckPoint
+from mindpet.graph.freeze_utils import freeze_modules, freeze_delta, freeze_from_config
+from mindpet.graph.ckpt_util import TrainableParamsCheckPoint
 
 __all__ = ['freeze_modules', 'freeze_delta', 'freeze_from_config', 'TrainableParamsCheckPoint']

tk/graph/ckpt_util.py → mindpet/graph/ckpt_util.py

@@ -15,9 +15,9 @@
 from mindspore.train.callback import ModelCheckpoint
 from mindspore.train.serialization import save_checkpoint, _get_merged_param_data
 
-from tk.utils.exceptions import AbsolutePathError, LinkPathError
-from tk.utils.io_utils import is_link_path
-from tk.log.log import logger
+from mindpet.utils.exceptions import AbsolutePathError, LinkPathError
+from mindpet.utils.io_utils import is_link_path
+from mindpet.log.log import logger
 
 
 class TrainableParamsCheckPoint(ModelCheckpoint):

tk/graph/freeze_utils.py → mindpet/graph/freeze_utils.py

@@ -10,10 +10,10 @@
 from fnmatch import fnmatch
 import mindspore.nn as nn
 
-from tk.log.log import logger
-from tk.utils.constants import DELTA_LIST
-from tk.utils.exceptions import ReadYamlFileError, ModelConfigFreezeInfoError, LinkPathError, UnexpectedError
-from tk.utils.io_utils import read_yaml_file
+from mindpet.log.log import logger
+from mindpet.utils.constants import DELTA_LIST
+from mindpet.utils.exceptions import ReadYamlFileError, ModelConfigFreezeInfoError, LinkPathError, UnexpectedError
+from mindpet.utils.io_utils import read_yaml_file
 
 
 def freeze_modules(model: nn.Cell,

tk/log/__init__.py → mindpet/log/__init__.py

@@ -2,6 +2,6 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-from tk.log.log import get_logger, logger, operation_logger_without_std, operation_logger, logger_without_std
+from mindpet.log.log import get_logger, logger, operation_logger_without_std, operation_logger, logger_without_std
 
 __all__ = ['get_logger', 'logger', 'logger_without_std', 'operation_logger', 'operation_logger_without_std']

tk/log/log.py → mindpet/log/log.py

@@ -10,15 +10,15 @@
 from typing import Dict, List, Tuple, Union
 import traceback
 
-from tk.utils.constants import DEFAULT_MAX_LOG_FILE_NUM, DEFAULT_MAX_LOG_FILE_SIZE, ABNORMAL_EXIT_CODE
-from tk.utils.exceptions import MakeDirError, UnsupportedPlatformError, PathOwnerError, PathModeError
+from mindpet.utils.constants import DEFAULT_MAX_LOG_FILE_NUM, DEFAULT_MAX_LOG_FILE_SIZE, ABNORMAL_EXIT_CODE
+from mindpet.utils.exceptions import MakeDirError, UnsupportedPlatformError, PathOwnerError, PathModeError
 
 try:
     from concurrent_log_handler import ConcurrentRotatingFileHandler as RFHandler
 except ImportError:
     from logging.handlers import RotatingFileHandler as RFHandler
 
-from tk.log.log_utils import check_list, const, convert_nodes_devices_input, create_dirs, \
+from mindpet.log.log_utils import check_list, const, convert_nodes_devices_input, create_dirs, \
     generate_rank_list, get_num_nodes_devices, get_rank_info, log_args_black_list_characters_replace, check_link_path
 
 logger_list = {}

tk/log/log_utils.py → mindpet/log/log_utils.py

@@ -7,8 +7,8 @@
 import platform
 from typing import Dict, List, Tuple, Union
 
-from tk.utils.exceptions import MakeDirError, UnsupportedPlatformError, PathOwnerError, PathModeError
-from tk.utils.constants import EMPTY_STRING
+from mindpet.utils.exceptions import MakeDirError, UnsupportedPlatformError, PathOwnerError, PathModeError
+from mindpet.utils.constants import EMPTY_STRING
 
 LOG_CONTENT_BLACK_LIST = ('\r', '\n', '\t', '\f', '\v', '\b', '\u000A', '\u000D', '\u000C',
                           '\u000B', '\u0008', '\u007F', '\u0009')

tk/security/param_check/base_check.py → mindpet/security/param_check/base_check.py

@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-from tk.security.param_check.option_check_utils import PathRightEscalationCheck, FileSizeCheck
+from mindpet.security.param_check.option_check_utils import PathRightEscalationCheck, FileSizeCheck
 
 
 class BaseCheckParam:

tk/security/param_check/model_config_params_check_util.py → mindpet/security/param_check/model_config_params_check_util.py

@@ -2,9 +2,9 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-from tk.log.log import logger
-from tk.utils.exceptions import ModelConfigParamsInfoError
-from tk.utils.constants import MODEL_CONFIG_PARAMS_CHARACTER_BLACKLIST, MODEL_CONFIG_LEN_LIMIT, \
+from mindpet.log.log import logger
+from mindpet.utils.exceptions import ModelConfigParamsInfoError
+from mindpet.utils.constants import MODEL_CONFIG_PARAMS_CHARACTER_BLACKLIST, MODEL_CONFIG_LEN_LIMIT, \
     TK_DEFINED_PARAM_NAME, INVALID_CUSTOM_PARAM_KEY_PREFIX, INVALID_CUSTOM_PARAM_VAL_PREFIX
 
 

tk/security/param_check/option_check_utils.py → mindpet/security/param_check/option_check_utils.py

@@ -3,12 +3,12 @@
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
 import os
-from tk.log.log import logger
-from tk.utils.io_utils import is_link_path
-from tk.utils.entrance_monitor import entrance_monitor
-from tk.utils.constants import EMPTY_STRING, BLACKLIST_CHARACTERS, BASE_WHITELIST_CHARACTERS, MAX_FILE_MB_SIZE, \
+from mindpet.log.log import logger
+from mindpet.utils.io_utils import is_link_path
+from mindpet.utils.entrance_monitor import entrance_monitor
+from mindpet.utils.constants import EMPTY_STRING, BLACKLIST_CHARACTERS, BASE_WHITELIST_CHARACTERS, MAX_FILE_MB_SIZE, \
     MB_SIZE, ENTRANCE_TYPE, MIN_MODE, MAX_MODE, GB_SIZE
-from tk.utils.exceptions import FileOversizeError, LinkPathError, LowDiskFreeSizeRiskError, PathContentError, \
+from mindpet.utils.exceptions import FileOversizeError, LinkPathError, LowDiskFreeSizeRiskError, PathContentError, \
     AbsolutePathError, PathLengthError, FileNameLengthError, PathGranularityError, PathOwnerError, PathModeError, \
     PathRightEscalationError, ManualCancelError
 

tk/task/evaluate_infer/evaluate_infer_task.py → mindpet/task/evaluate_infer/evaluate_infer_task.py

@@ -5,16 +5,16 @@
 import os
 import json
 import subprocess
-from tk.utils.io_utils import read_yaml_file, read_json_file
-from tk.security.param_check.option_check_utils import PathContentCheckParam
-from tk.task.evaluate_infer.result_file_check import ResultFileCheckParam, ResultFileCheck
-from tk.log.log import logger, operation_logger_without_std, record_operation_and_service_info_log, \
+from mindpet.utils.io_utils import read_yaml_file, read_json_file
+from mindpet.security.param_check.option_check_utils import PathContentCheckParam
+from mindpet.task.evaluate_infer.result_file_check import ResultFileCheckParam, ResultFileCheck
+from mindpet.log.log import logger, operation_logger_without_std, record_operation_and_service_info_log, \
     record_operation_and_service_error_log, record_operation_and_service_warning_log
-from tk.utils.constants import MODEL_CONFIG_ROOT_KEYS, EVAL_RESULT_FILE_NAME, INFER_RESULT_FILE_NAME, SPACE_CHARACTER, \
+from mindpet.utils.constants import MODEL_CONFIG_ROOT_KEYS, EVAL_RESULT_FILE_NAME, INFER_RESULT_FILE_NAME, SPACE_CHARACTER, \
     PATH_MODE_LIMIT, EVAL_INFER_TASK_NAMES, EVALUATE_TASK_NAME, EMPTY_STRING
-from tk.utils.task_utils import create_output_path_subdir_with_uuid, model_config_params_check_item, \
+from mindpet.utils.task_utils import create_output_path_subdir_with_uuid, model_config_params_check_item, \
     monitor_process_rsp_code, model_config_keys_check_item, path_in_out_loop_check
-from tk.utils.exceptions import LinkPathError, ReadYamlFileError, UnexpectedError, MonitorProcessRspError, \
+from mindpet.utils.exceptions import LinkPathError, ReadYamlFileError, UnexpectedError, MonitorProcessRspError, \
     CreateProcessError, TaskError, FileOversizeError, PathRightEscalationError
 
 

tk/task/evaluate_infer/result_file_check.py → mindpet/task/evaluate_infer/result_file_check.py

@@ -2,8 +2,8 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-from tk.security.param_check.base_check import BaseCheckParam, BaseCheck
-from tk.security.param_check.option_check_utils import PathExistCheck, LinkPathCheck, get_real_path
+from mindpet.security.param_check.base_check import BaseCheckParam, BaseCheck
+from mindpet.security.param_check.option_check_utils import PathExistCheck, LinkPathCheck, get_real_path
 
 
 class ResultFileCheckParam(BaseCheckParam):

tk/task/finetune/finetune_options_check.py → mindpet/task/finetune/finetune_options_check.py

@@ -2,9 +2,9 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-from tk.utils.constants import GB_SIZE
-from tk.security.param_check.base_check import BaseCheckParam, BaseCheck
-from tk.security.param_check.option_check_utils import PathContentBlacklistCharactersCheck, AbsolutePathCheck, \
+from mindpet.utils.constants import GB_SIZE
+from mindpet.security.param_check.base_check import BaseCheckParam, BaseCheck
+from mindpet.security.param_check.option_check_utils import PathContentBlacklistCharactersCheck, AbsolutePathCheck, \
     PathExistCheck, LinkPathCheck, PathContentLengthCheck, PathContentCharacterCheck, PathGranularityCheck, \
     DiskFreeSpaceCheck, get_real_path
 

tk/task/finetune/finetune_task.py → mindpet/task/finetune/finetune_task.py

@@ -4,13 +4,13 @@
 
 import os
 import subprocess
-from tk.utils.io_utils import read_yaml_file
-from tk.utils.constants import MODEL_CONFIG_ROOT_KEYS
-from tk.utils.exceptions import LinkPathError, TaskError, ReadYamlFileError, UnexpectedError, CreateProcessError, \
+from mindpet.utils.io_utils import read_yaml_file
+from mindpet.utils.constants import MODEL_CONFIG_ROOT_KEYS
+from mindpet.utils.exceptions import LinkPathError, TaskError, ReadYamlFileError, UnexpectedError, CreateProcessError, \
     MonitorProcessRspError
-from tk.log.log import record_operation_and_service_error_log, record_operation_and_service_info_log, \
+from mindpet.log.log import record_operation_and_service_error_log, record_operation_and_service_info_log, \
     operation_logger_without_std
-from tk.utils.task_utils import create_output_path_subdir_with_uuid, model_config_params_check_item, \
+from mindpet.utils.task_utils import create_output_path_subdir_with_uuid, model_config_params_check_item, \
     extend_model_config_freeze_command, monitor_process_rsp_code, model_config_keys_check_item, path_in_out_loop_check
 
 

tk/task/option_decorators.py → mindpet/task/option_decorators.py

@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-from tk.task.options import DataPathOption, PretrainedModelPathOption, OutputPathOption, BootFilePathOption, \
+from mindpet.task.options import DataPathOption, PretrainedModelPathOption, OutputPathOption, BootFilePathOption, \
     ModelConfigPathOption, QuietOption, CkptPathOption, TimeoutOption
 
 

mindpet/task/options/__init__.py

@@ -0,0 +1,12 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
+
+from mindpet.task.options.data_path_option import DataPathOption
+from mindpet.task.options.pretrained_model_path_option import PretrainedModelPathOption
+from mindpet.task.options.output_path_option import OutputPathOption
+from mindpet.task.options.boot_file_path_option import BootFilePathOption
+from mindpet.task.options.model_config_path_option import ModelConfigPathOption
+from mindpet.task.options.quiet_option import QuietOption
+from mindpet.task.options.ckpt_path_option import CkptPathOption
+from mindpet.task.options.timeout_option import TimeoutOption

tk/task/options/boot_file_path_option.py → mindpet/task/options/boot_file_path_option.py

@@ -5,9 +5,9 @@
 import os
 import click
 from click.exceptions import MissingParameter, BadParameter
-from tk.utils.constants import PATH_MODE_LIMIT
-from tk.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
-from tk.task.options.path_check_param import default_file_length_check_param, default_file_content_check_param
+from mindpet.utils.constants import PATH_MODE_LIMIT
+from mindpet.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
+from mindpet.task.options.path_check_param import default_file_length_check_param, default_file_content_check_param
 
 
 class BootFilePathOption(click.core.Option):

tk/task/options/ckpt_path_option.py → mindpet/task/options/ckpt_path_option.py

@@ -4,9 +4,9 @@
 
 import click
 from click.exceptions import MissingParameter
-from tk.utils.constants import PATH_MODE_LIMIT
-from tk.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
-from tk.task.options.path_check_param import default_folder_length_check_param, default_folder_content_check_param
+from mindpet.utils.constants import PATH_MODE_LIMIT
+from mindpet.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
+from mindpet.task.options.path_check_param import default_folder_length_check_param, default_folder_content_check_param
 
 
 class CkptPathOption(click.core.Option):

tk/task/options/data_path_option.py → mindpet/task/options/data_path_option.py

@@ -4,9 +4,9 @@
 
 import click
 from click.exceptions import MissingParameter
-from tk.utils.constants import PATH_MODE_LIMIT
-from tk.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
-from tk.task.options.path_check_param import default_folder_length_check_param, default_folder_content_check_param
+from mindpet.utils.constants import PATH_MODE_LIMIT
+from mindpet.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
+from mindpet.task.options.path_check_param import default_folder_length_check_param, default_folder_content_check_param
 
 
 class DataPathOption(click.core.Option):

tk/task/options/model_config_path_option.py → mindpet/task/options/model_config_path_option.py

@@ -4,9 +4,9 @@
 
 import click
 from click.exceptions import BadParameter
-from tk.utils.constants import PATH_MODE_LIMIT
-from tk.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
-from tk.task.options.path_check_param import default_file_length_check_param, default_file_content_check_param
+from mindpet.utils.constants import PATH_MODE_LIMIT
+from mindpet.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
+from mindpet.task.options.path_check_param import default_file_length_check_param, default_file_content_check_param
 
 
 class ModelConfigPathOption(click.core.Option):

tk/task/options/output_path_option.py → mindpet/task/options/output_path_option.py

@@ -4,9 +4,9 @@
 
 import click
 from click.exceptions import MissingParameter
-from tk.utils.constants import PATH_MODE_LIMIT
-from tk.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
-from tk.task.options.path_check_param import default_folder_length_check_param, default_folder_content_check_param
+from mindpet.utils.constants import PATH_MODE_LIMIT
+from mindpet.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
+from mindpet.task.options.path_check_param import default_folder_length_check_param, default_folder_content_check_param
 
 
 class OutputPathOption(click.core.Option):

tk/task/options/path_check_param.py → mindpet/task/options/path_check_param.py

@@ -2,9 +2,9 @@
 # -*- coding: utf-8 -*-
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
-from tk.utils.constants import SPACE_CHARACTER
-from tk.security.param_check.option_check_utils import PathLengthCheckParam, PathContentCheckParam
-from tk.utils.constants import DEFAULT_PATH_LEN_MIN_LIMIT, DEFAULT_PATH_LEN_MAX_LIMIT, DEFAULT_FILE_LEN_MIN_LIMIT, \
+from mindpet.utils.constants import SPACE_CHARACTER
+from mindpet.security.param_check.option_check_utils import PathLengthCheckParam, PathContentCheckParam
+from mindpet.utils.constants import DEFAULT_PATH_LEN_MIN_LIMIT, DEFAULT_PATH_LEN_MAX_LIMIT, DEFAULT_FILE_LEN_MIN_LIMIT, \
     DEFAULT_FILE_LEN_MAX_LIMIT
 
 default_folder_length_check_param = PathLengthCheckParam(path_min_limit=DEFAULT_PATH_LEN_MIN_LIMIT,

tk/task/options/pretrained_model_path_option.py → mindpet/task/options/pretrained_model_path_option.py

@@ -3,9 +3,9 @@
 # Copyright © Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 
 import click
-from tk.utils.constants import PATH_MODE_LIMIT
-from tk.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
-from tk.task.options.path_check_param import default_folder_length_check_param, default_folder_content_check_param
+from mindpet.utils.constants import PATH_MODE_LIMIT
+from mindpet.task.finetune.finetune_options_check import FinetuneOptionsCheckParam, FinetuneOptionsCheck
+from mindpet.task.options.path_check_param import default_folder_length_check_param, default_folder_content_check_param
 
 
 class PretrainedModelPathOption(click.core.Option):

tk/task/options/timeout_option.py → mindpet/task/options/timeout_option.py

@@ -5,7 +5,7 @@
 import re
 import click
 from click.exceptions import BadParameter
-from tk.utils.constants import TIMEOUT_REGEX, HOURS_PER_DAY, SPACE_CHARACTER, EMPTY_STRING, SECONDS_PER_HOUR
+from mindpet.utils.constants import TIMEOUT_REGEX, HOURS_PER_DAY, SPACE_CHARACTER, EMPTY_STRING, SECONDS_PER_HOUR
 
 
 class TimeoutOption(click.core.Option):

tk/tk_main.py → mindpet/tk_main.py

@@ -4,13 +4,13 @@
 
 import sys
 import click
-from tk.utils.task_utils import handle_exception_log
-from tk.utils.entrance_monitor import entrance_monitor
-from tk.task.finetune.finetune_task import FinetuneTask
-from tk.task.evaluate_infer.evaluate_infer_task import EvaluateInferTask
-from tk.log.log import set_logger_property, record_operation_and_service_info_log
-from tk.task.option_decorators import finetune_options, evaluate_options, infer_options
-from tk.utils.constants import ENTRANCE_TYPE, EVALUATE_TASK_NAME, INFER_TASK_NAME, ABNORMAL_EXIT_CODE
+from mindpet.utils.task_utils import handle_exception_log
+from mindpet.utils.entrance_monitor import entrance_monitor
+from mindpet.task.finetune.finetune_task import FinetuneTask
+from mindpet.task.evaluate_infer.evaluate_infer_task import EvaluateInferTask
+from mindpet.log.log import set_logger_property, record_operation_and_service_info_log
+from mindpet.task.option_decorators import finetune_options, evaluate_options, infer_options
+from mindpet.utils.constants import ENTRANCE_TYPE, EVALUATE_TASK_NAME, INFER_TASK_NAME, ABNORMAL_EXIT_CODE
 
 
 def cli_wrapper():