in weak_to_strong/datasets.py [0:0]
def register_dataset(name: str, config: DatasetConfig): _REGISTRY[name] = config