4445f32798d72e291eac7661657092b06d7b9326,rllib/env/pettingzoo_env.py,ParallelPettingZooEnv,step,#ParallelPettingZooEnv#Any#,236

Before Change

    def step(self, action_dict):
        for agent in self.agents:
            action_dict[agent] = self.action_space.sample()
        obs, rew, dones, info = self.par_env.step(action_dict)
        dones["__all__"] = all(dones.values())
        return obs, rew, dones, info

After Change

        aobs, arew, adones, ainfo = self.par_env.step(action_dict)
        obss = {}
        rews = {}
        dones = {}
        infos = {}
        for agent in action_dict:
            obss[agent] = aobs[agent]
            rews[agent] = arew[agent]

Instances