Method update_record

eval_heldout/hotpotQA/src/wrappers.py:235–238

(self)

    return obs, reward, done, info

  def update_record(self):
    if len(self.traj) > 0:
      self.trajs.append(self.traj)
      self.traj = {"observations": [], "actions": []}
  
  def write(self):
    self.update_record()