def chunk()

in src/pipeline/prepare.py [0:0]

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

def chunk(x, batch_size):
    """split array into chunks of batch_size
    """
    chunk_range = range(0, x.shape[0], batch_size)
    chunks = [x[p: p + batch_size] for p in chunk_range]
    return chunks