Python split generator into sub lists

chunks.py

def chunks(generator, chunk_size):
    """Yield successive chunks from a generator"""
    chunk = []

    for item in generator:
        if len(chunk) >= chunk_size:
            yield chunk
            chunk = [item]
        else:
            chunk.append(item)

    if chunk:
        yield chunk

eg.

>>> for chunk in chunks(range(10), 4):
...       print(chunk)
...
[0, 1, 2, 3]
[4, 5, 6, 7]
[8, 9]