def get_predictions(decode_steps):
  """Returns predictions dict given DecodeSteps."""
  return dict(
      predicted_action_types=decode_steps.action_types,
      predicted_action_ids=decode_steps.action_ids)