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

Before Change

        
        env_done = False
        // iterate over self.agents
        for agent in self.agents:

            // Execute only for agents that have not been done in previous steps
            if agent in action_dict.keys():
                if not env_done:
                    assert agent == self.aec_env.agent_selection, \
                        f"environment has a nontrivial ordering, and " \
                        "cannot be used with the POMGameEnv wrapper\"" \
                        "nCurrent agent: {self.aec_env.agent_selection}" \
                        "\nExpected agent: {agent}"
                    // Execute agent action in environment
                    self.obs[agent] = self.aec_env.step(
                        action_dict[agent], observe=True)
                    if all(self.aec_env.dones.values()):
                        env_done = True
                        self.dones["__all__"] = True
                else:
                    self.obs[agent] = self.aec_env.observe(agent)
                // Get reward
                self.rewards[agent] = self.aec_env.rewards[agent]
                // Update done status
                self.dones[agent] = self.aec_env.dones[agent]

            // For agents with done = True, remove from dones, rewards and
            // observations.
            else:
                del self.dones[agent]
                del self.rewards[agent]
                del self.obs[agent]
                del self.infos[agent]

        // update self.agents
        self.agents = list(action_dict.keys())

        // Update infos stepwise
        for agent in self.agents:

After Change

                "__all__" (required) is used to indicate env termination.
            infos (dict): Optional info values for each agent id.
        
        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"

Instances