ValueError(“None values not supported.”) tensorflow 2.2.0, sonnet 2.0.0b0

Question

Good People of the world,

All related files are attached with the Github link: https://github.com/sandeepsinghsengar/PMPU

I am getting the following error, als