Fix or ignore type errors generated by the next release of pytype.
Fix or ignore type errors generated by the next release of pytype.

PiperOrigin-RevId: 315827993

tensorflow/data-validation,tensorflow/data-validation,tensorflow/data-validation