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

Before Change

                        "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:

After Change

            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]
                   for agent in action_dict), \

Instances