Fix minor bug in TerminateIllegal wrapper indexing empty info dict (#…

…1129)
Farama-Foundation · Nov 7, 2023 · eaf6693 · eaf6693
1 parent defe0d1
commit eaf6693
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pettingzoo/utils/wrappers/terminate_illegal.py b/pettingzoo/utils/wrappers/terminate_illegal.py
@@ -31,7 +31,10 @@ def observe(self, agent: AgentID) -> ObsType | None:
         obs = super().observe(agent)
         if agent == self.agent_selection:
             self._prev_obs = obs
-            self._prev_info = self.infos[self.agent_selection]
+            if self.agent_selection in self.infos:
+                self._prev_info = self.infos[self.agent_selection]
+            else:
+                self._prev_info = {}
         return obs
 
     def step(self, action: ActionType) -> None: