numpy.AxisError: axis 1 is out of bounds for array of dimension 1?

Question

    def prepare_data_cnn(self):
        assert self.args.data_mode == \'sum\', \\
                \'Use [-m sum] for training CNN\'

        self.train_X = tf.keras.