In preparation for cl/296449699, disabling some pytype checks that fail after this CL is submitted.
In preparation for cl/296449699, disabling some pytype checks that fail after
this CL is submitted.

The root cause for these failures is that currently tensorflow.compat.v1 maps
to Any and effectively there isn't any type checking. cl/296449699 enables
this type checking for tensorflow.compat.v1 and as a result exposes some bugs /
problems with the type checking code that got checked in previously.

PiperOrigin-RevId: 297331371
Change-Id: I7bb4ccb53ef6aebcae69e809ce245b1f02882ea9

deepmind/sonnet,deepmind/sonnet