Small datagen correction to only exclude parsing problems that need the flag.
Small datagen correction to only exclude parsing problems that need the flag.

PiperOrigin-RevId: 201056385

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