SRHarness / src /sr_harness /utils /__init__.py
yuzihan@SIM1
Deploy SRHarness Docker Space
d06fa57
Raw History Blame Contribute Delete
1.51 kB
# Copyright (c) 2024-present, Yumeow. Licensed under the MIT License.
from .attr_dict import *
from .logger import *
from .metrics import *
from .timing import *
from .utils import *
from .tag2ansi import tag2ansi
from .save_args import save_args
from .log_exception import log_exception
from .classproperty import classproperty
from .render_python import render_python
from .symbolic_acc import get_symbolic_acc
from .render_markdown import render_markdown
from .factory_mixin import FactoryMixin
from .fix_parser import add_minus_flags, add_negation_flags
from .lazy_loader import setup_lazy_imports, TYPE_CHECKING
from .df_to_3line import df_to_3line
from .sanitize_filename import sanitize_filename
from .format_confusion_matrix import format_confusion_matrix
from .parse_json_with_template import parse_json_with_template
from .model_store import get_default, download_model, upload_model
from .format_pareto_front import format_pareto_front
from .constant_optimizer import ConstantOptimizerConfig, fit_constants
# 引入可选依赖的子模块
if TYPE_CHECKING:
from . import plot
from .auto_gpu import AutoGPU
from . import nn
__getattr__, __dir__, __all__ = setup_lazy_imports(__name__, {
"plot": (".plot", "all"),
"AutoGPU": (".auto_gpu", "nn"), # 可选引入 .auto_gpu.AutoGPU, 但需要通过 pip install sr-harness[nn] 来安装可选依赖
"nn": (".nn", "nn"), # 将 .nn package 映射到 nn,并标明需要通过 pip install sr-harness[nn] 来安装可选依赖
})