code: def _ExtractTFLitePredictions(  # pylint: disable=invalid-name
      extracts: beam.pvalue.PCollection,
      eval_config: config_pb2.EvalConfig,
      eval_shared_models: Dict[str, types.EvalSharedModel],
      do_fn: Callable[..., TFLitePredictionDoFn],
  ) -> beam.pvalue.PCollection:
      """A PTransform that adds predictions and possibly other tensors to extracts.
  
      Args:
        extracts: PCollection of extracts containing model inputs keyed by
          tfma.FEATURES_KEY.
        eval_config: Eval config.
        eval_shared_models: Shared model parameters keyed by model name.
        do_fn: Constructor for TFLitePredictionDoFn.
  
      Returns:
        PCollection of Extracts updated with the predictions.
      """
      return extracts | "Predict" >> beam.ParDo(
          do_fn(
              eval_config=eval_config,
              eval_shared_models=eval_shared_models,
          )
      )
  
imports: import collections
  import copy
  from typing import Callable, Dict, Sequence, Union
  from absl import logging
  import apache_beam
  import numpy
  import tensorflow
  from tensorflow_model_analysis import constants
  from tensorflow_model_analysis.api import types
  from tensorflow_model_analysis.extractors import extractor
  from tensorflow_model_analysis.proto import config_pb2
  from tensorflow_model_analysis.utils import model_util
  from tensorflow_model_analysis.utils import util
label_desc: Use image node_id 1 for calling a global function with example usage: _ExtractTFLitePredictions(extracts, eval_config, eval_shared_models, do_fn) and returns: unknown