Visualization issues related to C2W4_Assignment

Hi everyone,

I have a strange issue with C2W4_Assignment, when visualizing the images on my laptop, they look fine:

But when I use the same code on colab, it goes like this:

I don’t know what’s going on…?

I have found the solution, the return statement was not indented properly and was having wrong values of images and labels.