Function get_model

fmpose3d/models/base_model.py:54–64

Return the model class registered under *name*. Raises :class:`KeyError` with a helpful message when the name is unknown.

(name: str)

def get_model(name: str) -> type["BaseModel"]:
    """Return the model class registered under *name*.

    Raises :class:`KeyError` with a helpful message when the name is unknown.
    """
    if name not in _MODEL_REGISTRY:
        available = ", ".join(sorted(_MODEL_REGISTRY)) or "(none)"
        raise KeyError(
            f"Unknown model '{name}'. Available models: {available}"
        )
    return _MODEL_REGISTRY[name]


def list_models() -> list[str]: