def gen_data(num_items): index = np.arange(num_items) dataset = Dataset(index=index, batch_class=ArrayBatch) return dataset