code: def _get_compression_proto(
      compression,
  ) -> data_service_pb2.DataServiceMetadata.Compression:
      if compression == COMPRESSION_AUTO:
          return data_service_pb2.DataServiceMetadata.COMPRESSION_SNAPPY
      if compression == COMPRESSION_NONE:
          return data_service_pb2.DataServiceMetadata.COMPRESSION_OFF
      raise ValueError(
          f"Invalid `compression` argument: {compression}. "
          f"Must be one of {[COMPRESSION_AUTO, COMPRESSION_NONE]}."
      )
  
imports: import enum
  import functools
  from typing import Callable
  from tensorflow.core.protobuf import data_service_pb2
  from tensorflow.python import tf2
  from tensorflow.python.data.experimental.ops import compression_ops
  from tensorflow.python.data.experimental.service import _pywrap_server_lib
  from tensorflow.python.data.experimental.service import _pywrap_utils
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import options
  from tensorflow.python.data.ops import structured_function
  from tensorflow.python.data.ops.options import AutoShardPolicy
  from tensorflow.python.data.ops.options import ExternalStatePolicy
  from tensorflow.python.eager import context
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import tensor_util
  from tensorflow.python.ops import gen_experimental_dataset_ops
  from tensorflow.python.ops import string_ops
  from tensorflow.python.saved_model import nested_structure_coder
  from tensorflow.python.util.tf_export import tf_export
label_desc: Use image node_id 4 for calling a global function with example usage: _get_compression_proto(compression) and returns: data_service_pb2, data_service_pb2