W2_A2_Ex-1_expected output error

anyone who can guide how to solve this as i am not excellent in python but i am good .
stuck in it …
if yes please explain with logic thanks

Why are you using a different index to the shape for the m_test case versus the m_train case? The organization of both datasets is the same.

Try uncommenting those additional print statements to see the full shapes of all the arrays and that should make it more clear.