Fix open-source test for MADE.
Fix open-source test for MADE.

PiperOrigin-RevId: 225467601

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