explicitly return observe

phonx · May 10, 2017 · 6812f3f · 6812f3f
1 parent 3190b97
commit 6812f3f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/parlai/agents/drqa/agents.py b/parlai/agents/drqa/agents.py
@@ -149,6 +149,7 @@ def observe(self, observation):
             observation['text'] = '\n'.join(dialogue)
         self.observation = observation
         self.episode_done = observation['episode_done']
+        return self.observation
 
     def act(self):
         """Update or predict on a single example (batchsize = 1)."""

diff --git a/parlai/core/worlds.py b/parlai/core/worlds.py
@@ -460,8 +460,8 @@ def batch_observe(self, index, batch_actions):
         batch_observations = []
         for i, w in enumerate(self.worlds):
             agents = w.get_agents()
-            agents[index].observe(validate(batch_actions[i]))
-            batch_observations.append(agents[index].observation)
+            observation = agents[index].observe(validate(batch_actions[i]))
+            batch_observations.append(observation)
         return batch_observations
 
     def batch_act(self, index, batch_observation):