f2408b719c991b98ba5dc77db81dc184805b33df,rllib/env/pettingzoo_env.py,PettingZooEnv,step,#PettingZooEnv#Any#,141

Before Change

            // observations.
            else:
                del self.dones[agent]
                del self.rewards[agent]
                del self.obs[agent]
                del self.infos[agent]

        // update self.agents

After Change

        stepped_agents = set()
        while self.aec_env.agent_selection not in stepped_agents:
            agent = self.aec_env.agent_selection
            assert agent in action_dict, \
                "Live environment agent is not in actions dictionary"
            self.aec_env.step(action_dict[agent])
            stepped_agents.add(agent)

        assert all(agent in stepped_agents or self.aec_env.dones[agent]

Instances