in weak_to_strong/datasets.py [0:0]
def hf_loader(*hf_name, split_names=None): if split_names is None: split_names = dict() return lambda split: hf_load_dataset(*hf_name, split=split_names.get(split, split))