def METHOD_NAME():
    batch_data = []
    batch_label = []
    for sample in reader():
        image = np.array(sample[0]).reshape(1, 28, 28)
        label = np.array(sample[1]).reshape(1)
        batch_data.append(image)
        batch_label.append(label)
        if len(batch_data) == 128:
            batch_data = np.array(batch_data, dtype='float32')
            batch_label = np.array(batch_label, dtype='int64')
            yield [batch_data, batch_label]
            batch_data = []
            batch_label = []