TensorFlow 2 in Google Colabs Failed Preconditioned Error on Model Test

Question

I am trying to call predict() on my full model and passing it the test data batch. The model works on my training and validation data, but I get this error when I try to run