I was running this tutorial on google colab with TF2.3 and this command
rotate = tfa.image.rotate(img, tf.constant(np.pi/8))
under "Rotate" section fails to work with the following error message:

Can someone tell me how to fix this? Thank you very much!