TypeError: ufunc 'isfinite' not supported for the input types

[ashikshafi08]

Computer Vision and Convolutional Neural Networks in TensorFlow

I was working on the video lecture named Becoming One With Data Part 2 when we create a function called view_random_image() where we will pass a directory and the class name and the function return us back an image.

I got a weird error attached the picture below,

The fishy thing is the expected output is returned but yet the error stays. I even copy-pasted and ran just that code snippet from the course notebook and still getting the same error.
But during the video lecture, this error didn't show up everything was smooth. Should we really care about this error?
@mrdbourke

[ashikshafi08]

Well nvm 😅 commenting out the plt.axes('off') worked, but weird Idk why this happened.

Solution

To turn off the axes, plt.axis('off); doesn't throw an error and everything works smooth!

[mrdbourke]

Hmmm. That is strange, I haven't seen that before!

Glad you got it fixed though.

I just checked the matplotlib docs, looks like False and "off" should both work: https://matplotlib.org/stable/api/_as_gen/matplotlib.pyplot.axis.html