def hf_loader()

in weak_to_strong/datasets.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

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))