fix 4 single turn
@@ -70,9 +70,9 @@ class LocalHumanAgent(Agent):
return {'episode_done': True}
reply_text = reply_text.replace('\\n', '\n')
- if self.opt.get('single_turn', False):
- reply_text += '[DONE]'
reply['episode_done'] = False
+ if self.opt.get('single_turn', False):
+ reply.force_set('episode_done', True)
reply['label_candidates'] = self.fixedCands_txt
if '[DONE]' in reply_text:
# let interactive know we're resetting
