While sampling in `env_problem_utils.py` cast up to `np.float64` This seems to fix things.
While sampling in `env_problem_utils.py` cast up to `np.float64`
This seems to fix things.

PiperOrigin-RevId: 248021102

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