C2W4_assigment - create parse_data_from_input function

I get an error when I try to reshape my data in the create parse_data_from_input function. I used np.reshape() after splitting the dataset into labels and images. What could possibly be wrong?

parse_data_from_input should be capable of handling both training and validation files. Use the number of rows of data you’ve processed when reshaping the image array.