Removing renaming of transformed features to simplify code.  Also some small cleanup.
Removing renaming of transformed features to simplify code.  Also some small cleanup.

PiperOrigin-RevId: 401289565

tensorflow/tfx,tensorflow/tfx