fixed classic observe bug

rfali · Apr 1, 2021 · 9603819 · 9603819
1 parent 3e856cf
commit 9603819
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pettingzoo/utils/wrappers/terminate_illegal.py b/pettingzoo/utils/wrappers/terminate_illegal.py
@@ -22,7 +22,8 @@ def reset(self):
 
     def observe(self, agent):
         obs = super().observe(agent)
-        self._prev_obs = obs
+        if agent == self.agent_selection:
+            self._prev_obs = obs
         return obs
 
     def step(self, action):