import tensorflow as tf


    return tf.estimator.EstimatorSpec(
        tf.estimator.ModeKeys.PREDICT,
        predictions=predictions,
        export_outputs={
            "output": tf.estimator.export.PredictOutput(export_out)
        })

  def _normalize_body_output(self, body_out):
    if isinstance(body_out, tuple):
