Modify run_env in dql_grasping_lib to support TF-Agents environments.
Modify run_env in dql_grasping_lib to support TF-Agents environments.

Reward summaries are also now only written by the 0th task. This is done to reduce the number of summaries saved in many-replica jobs, to speed up Tensorboard processing.

PiperOrigin-RevId: 264520440

google-research/tensor2robot