def _override_parent_methods(keep_items):
  """Makes inheritted attributes of the TFT layer unusable and undocumented."""
  for name in dir(tf_keras.Model):
    if name.startswith('_') or name in keep_items:
      continue
    if callable(getattr(tf_keras.Model, name)):
      setattr(TransformFeaturesLayer, name, _make_method_override(name))
    elif not isinstance(getattr(TransformFeaturesLayer, name), property):
      doc_controls.do_not_generate_docs(getattr(TransformFeaturesLayer, name))
Makes inheritted attributes of the TFT layer unusable and undocumented.
tensorflow_transform/output_wrapper.py