Tensorflow 2.2.0 compatibility

[JenineJ]

Thanks for putting this together!
Have you been able to use this with Tensorflow 2.2.0? It worked fine for me in 2.1.0, but isn't working now that I updated Tensorflow.