TensorFlow 2.1 - TypeError: padded_batch() missing 1 required positional argument: 'padded_shapes'

Question

I am trying to replicate https://keras.io/examples/vision/retinanet/ tutorial on my Local environment where I have CUDA. But, since I am using Windows, TensorFlow version is