def batched()

in src/open_r1/utils/competitive_programming/utils.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def batched(iterable, n):
    "Batch data into lists of length n. The last batch may be shorter."
    # batched('ABCDEFG', 3) --> ABC DEF G
    if n < 1:
        return iterable
    it = iter(iterable)
    while batch := list(islice(it, n)):
        yield batch