def make_batches()

in src/util.py [0:0]

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

def make_batches(dataset, batch_size):
    batches = []
    cur_batch = []
    for i, ex in enumerate(dataset):
        cur_batch.append(ex)
        if len(cur_batch) == batch_size or i == len(dataset) - 1:
            batches.append(cur_batch)
            cur_batch = []
    return batches