same fix on line 1097

facebookresearch · Mar 10, 2021 · 6eec738 · 6eec738
1 parent 0b4e940
commit 6eec738
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parlai/core/worlds.py b/parlai/core/worlds.py
@@ -1094,7 +1094,7 @@ def parley(self):
             # we log the task act and the index of the act
             # in the buffer for world logging purposes
             self._task_acts[i] = act  # for world logging
-            self._task_acts[i]['dyn_batch_idx'] = i
+            self._task_acts[i].force_set('dyn_batch_idx', i)
 
             obs = self.worlds[i].get_model_agent().observe(act)
             self._obs[i] = obs