def invoke()

in src/chug/hfds/collate.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def invoke(f, *args, **kwargs):
    if isinstance(f, (IterableDataset, DataLoader)) and len(args) == 0:
        return iter(f)
    if isinstance(f, list):
        return iter(f)
    if callable(f):
        result = f(*args, **kwargs)
        return result
    raise ValueError(f"{f}: not a valid pipeline stage")