def register_dataset()

in weak_to_strong/datasets.py [0:0]


def register_dataset(name: str, config: DatasetConfig):
    _REGISTRY[name] = config