Rollback to match current stable tf.agents.
Rollback to match current stable tf.agents.

PiperOrigin-RevId: 346804167
Change-Id: Ic3a2dadb1825a18a71193effcc60b3a6ea5dc0dd

tensorflow/agents,tensorflow/agents