diff --git a/environment.py b/environment.py index 68bfebb38..5af3018b2 100644 --- a/environment.py +++ b/environment.py @@ -462,6 +462,7 @@ def end_round(self): # Clean up survivors for a in self.active_agents: a.add_event(e.SURVIVED_ROUND) + a.store_game_state(self.get_state_for_agent(a)) # Send final event to agents that expect them for a in self.agents: