Model Registry

get_model_registry_config()

Given a registry type and unique model identifier, build a config.

def get_model_registry_config(
    registry: ModelRegistryEnum,
    repo: str
)

registry

ModelRegistryEnum

required

The model registry type.

repo

str

required

The repository string to parse. For ModelRegistryEnum.HUGGINGFACE, this is a repo id. For ModelRegistryEnum.MLM, it is a uri to a model.

return

ModelRegistryConfig

The constructed ModelRegistryConfig object.