batched generator for Python

batched_generator.py

import itertools

def batch_generator(size, g):
    buffer = []
    try:
        while True:
            for i in range(0, size):
                buffer.append(g.next())
            yield buffer
            buffer = []
    except StopIteration:
        if 0 == len(buffer):
            raise
        else:
            yield buffer

@danbst thank you!