def METHOD_NAME(sample_info):
    idx_in_batch = sample_info.idx_in_batch
    size = sizes[idx_in_batch % len(sizes)]
    num_channel = num_channels[idx_in_batch % len(num_channels)]
    shape = (size[0], size[1], num_channel)
    return np.uint8(rng.uniform(0, 255, shape))