removing [DONE] from text
@@ -15,7 +15,7 @@ class LocalHumanAgent(Agent):
def __init__(self, opt, shared=None):
super().__init__(opt)
self.id = 'localHuman'
- self.done = False
+ self.episodeDone = False
def observe(self, msg):
print(display_messages([msg]))
@@ -27,9 +27,10 @@ class LocalHumanAgent(Agent):
reply_text = input("Enter Your Reply: ")
if '[DONE]' in reply_text:
reply['episode_done'] = True
- self.done = True
+ self.episodeDone = True
+ reply_text = reply_text.replace('[DONE]', '')
reply['text'] = reply_text
return reply
- def epoch_done(self):
- return self.done
+ def episode_done(self):
+ return self.episodeDone
