from ..runtime.factory import Registry

DATASET_REGISTRY = Registry('DATASET')


def build_dataset(_type: str, *args, **kwargs) -> object:
    return DATASET_REGISTRY.get(_type)(*args, **kwargs)


def list_datasets():
    return list(DATASET_REGISTRY._obj_map.keys())