code: def wrap_inference_results(inference_result_proto):
      """Returns packaged inference results from the provided proto.
  
      Args:
        inference_result_proto: The classification or regression response proto.
  
      Returns:
        An InferenceResult proto with the result from the response.
      """
      inference_proto = inference_pb2.InferenceResult()
      if isinstance(
          inference_result_proto,
          classification_pb2.ClassificationResponse,
      ):
          inference_proto.classification_result.CopyFrom(
              inference_result_proto.result
          )
      elif isinstance(
          inference_result_proto, regression_pb2.RegressionResponse
      ):
          inference_proto.regression_result.CopyFrom(
              inference_result_proto.result
          )
      return inference_proto
  
imports: from __future__ import division
  from __future__ import print_function
  import collections
  import copy
  import json
  import math
  from absl import logging
  import numpy
  import tensorflow
  from google.protobuf import json_format
  from six import binary_type, string_types, integer_types
  from six import iteritems
  from six.moves import zip
  from inspect import signature
  from utils import common_utils
  from utils import platform_utils
  from tensorflow_serving.apis import classification_pb2
  from tensorflow_serving.apis import inference_pb2
  from tensorflow_serving.apis import regression_pb2
label_desc: Use image node_id 4 for calling a global function with example usage: wrap_inference_results(inference_result_proto) and returns: inference_proto