import tensorflow as tf


  def body(self, features):
    observations = features["inputs"]
    obs_shape = observations.shape.as_list()
    # Just so Saver doesn't complain because of no variables.
    tf.get_variable("dummy_var", initializer=0.0)
    num_actions = self.hparams.problem.num_actions
    logits = tf.constant(
        1. / float(num_actions),
        shape=(obs_shape[:1] + [1, num_actions])
