def batch_data(self, rnd_array):
    index_data = []
    for i in range(7):
        index_data.extend(([i] * rnd_array[i]))
    return (rnd_array, index_data)
def batch_data(self, rnd_array):
    index_data = []
    for i in range(7):
        index_data.extend(([i] * rnd_array[i]))
    return (rnd_array, index_data)