Fix Travis breakage for tf != nightly.
Fix Travis breakage for tf != nightly.

PiperOrigin-RevId: 239653689

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