code: def simple_fixed_prediction_estimator(
      export_path,
      eval_export_path,
      output_prediction_key=prediction_keys.PredictionKeys.PREDICTIONS,
  ):
      estimator_metadata = (
          get_simple_fixed_prediction_estimator_and_metadata(
              output_prediction_key
          )
      )
      estimator_metadata["estimator"].train(
          input_fn=estimator_metadata["train_input_fn"], steps=1
      )
      return util.export_model_and_eval_model(
          estimator=estimator_metadata["estimator"],
          serving_input_receiver_fn=estimator_metadata[
              "serving_input_receiver_fn"
          ],
          eval_input_receiver_fn=estimator_metadata[
              "eval_input_receiver_fn"
          ],
          export_path=export_path,
          eval_export_path=eval_export_path,
      )
  
imports: import tensorflow
  from tensorflow import estimator
  from tensorflow_model_analysis.eval_saved_model import export
  from tensorflow_model_analysis.eval_saved_model.example_trainers import util
  from tensorflow.python.estimator.canned import metric_keys
  from tensorflow.python.estimator.canned import prediction_keys
label_desc: Use image node_id 2 for calling a global function with example usage: simple_fixed_prediction_estimator(export_path, eval_export_path, output_prediction_key) and returns: util