UniRig/src/model/parse.py

from .unirig_ar import UniRigAR
from .unirig_skin import UniRigSkin

from .spec import ModelSpec

def get_model(**kwargs) -> ModelSpec:
    MAP = {
        'unirig_ar': UniRigAR,
        'unirig_skin': UniRigSkin,
    }
    __target__ = kwargs['__target__']
    del kwargs['__target__']
    assert __target__ in MAP, f"expect: [{','.join(MAP.keys())}], found: {__target__}"
    return MAP[__target__](**kwargs)