Fix invalid type annotations.
Fix invalid type annotations.

PiperOrigin-RevId: 400166400
Change-Id: Ic8cf848f29863686ab0cb6e36f65ffb759f28e12

deepmind/dm_robotics,deepmind/dm_robotics,deepmind/dm_robotics