code: def feature_skew_sink(
      output_path_prefix: str, proto: Type[message.Message]
  ) -> beam.PTransform:
      """Sink for writing feature skew results."""
      return beam.io.WriteToTFRecord(
          output_path_prefix, coder=beam.coders.ProtoCoder(proto)
      )
  
imports: from typing import Callable, Iterable, Iterator, Optional, Type, TypeVar
  import apache_beam
  import tensorflow
  from google.protobuf import message
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 4 for calling a global function with example usage: feature_skew_sink(output_path_prefix, proto) and returns: beam