Prepare for upcoming keras initializer change.
Prepare for upcoming keras initializer change.

PiperOrigin-RevId: 453457229
Change-Id: Id8cd4c6409876d4d27fec03da8ab88f68ad17c26

tensorflow/agents,tensorflow/agents