def full_modalities(hparams):
  """Full modalities with actions and rewards."""
  hparams.problem_hparams.input_modality = {
      "inputs": ("video:l2raw", 256),
      "input_reward": ("symbol:one_hot", 3),
      "input_action": ("symbol:one_hot", 5)
  }
  hparams.problem_hparams.target_modality = {
      "targets": ("video:l2raw", 256),
      "target_reward": ("symbol:one_hot", 3),
      "target_action": ("symbol:one_hot", 5),
  }
  hparams.force_full_predict = True
  return hparams