update the reacher env test.
update the reacher env test.

PiperOrigin-RevId: 253316740

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor