def batched_get()

in understanding_rl_vision/rl_clarity/interface.py [0:0]

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

def batched_get(data, batch_size, process_minibatch):
    n_points = data.shape[0]
    n_minibatches = -((-n_points) // batch_size)
    return np.concatenate(
        [
            process_minibatch(data[i * batch_size : (i + 1) * batch_size])
            for i in range(n_minibatches)
        ],
        axis=0,
    )