"TypeError: Invalid shape (28, 28, 1) for image data" in SageMaker_AIShield_Integration.ipynb #23
Description
If I run SageMaker_AIShield_Integration.ipynb in SageMaker Studio default Data Science (Python 3) kernel, I get the error in the cell "Visualize few random data sample":
TypeError: Invalid shape (28, 28, 1) for image data