add back 'trajectory' field in outputs

ronghanghu · May 16, 2018 · 593a6d1 · 593a6d1
1 parent 684c54f
commit 593a6d1
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tasks/R2R/rational_follower.py b/tasks/R2R/rational_follower.py
@@ -151,7 +151,6 @@ def run_rational_follower(
  if 'attentions' in candidate:
  candidate['attentions'] = [list(tens) for tens in candidate['attentions']]
  del candidate['instr_encoding']
- del candidate['trajectory']
  candidate['rank'] = i
  candidate['gold'] = (include_gold and i == 0)
  utils.pretty_json_dump(candidate_lists_by_instr_id, f)