defect-analysis/defect_analysis/ml/model_registry.py

"""模型后端检测与注册信息。"""

import importlib.util


def detect_optional_model_backends():
    """检测可选 ML 后端是否可用，缺失时不影响主应用启动。"""
    return {
        "xgboost": importlib.util.find_spec("xgboost") is not None,
        "lightgbm": importlib.util.find_spec("lightgbm") is not None,
        "torch": importlib.util.find_spec("torch") is not None,
        "tensorflow": importlib.util.find_spec("tensorflow") is not None,
    }