At present, I am working on an already existing code (developed using Tensorflow 1.10.0 and sonnet 1.35). Trying to convert this code for Tf 2.2.0 and sonnet 2.0.0. However,