Ttius's picture
Upload 192 files
998bb30 verified
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())