from uniperceiver.utils.registry import Registry

EMBEDDING_REGISTRY = Registry("EMBEDDING")
EMBEDDING_REGISTRY.__doc__ = """
Registry for embedding
"""

def build_embeddings(cfg, name):
    embeddings = None if name.lower() == "none" else EMBEDDING_REGISTRY.get(name)(cfg)
    return embeddings