code: def fix_internal_object_pickling():
      """Fix pickling issues (see b/121323638)."""
      for proto_cls in _PROTO_CLASSES:
          copyreg.pickle(proto_cls, _pickle_proto)
  
      copyreg.pickle(tf.TensorSpec, _pickle_tensor_spec)
  
file_path: transform/tensorflow_transform/pickle_helper.py
imports: import copyreg
  import tensorflow
  from tensorflow_transform import common
  from tensorflow_metadata.proto.v0 import schema_pb2
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 5 for calling a global function with example usage: fix_internal_object_pickling() without return types