1999266bbaa82eac33a06b0b0a66821f14d77936,rllib/env/pettingzoo_env.py,PettingZooEnv,reset,#PettingZooEnv#,117

Before Change

        for agent in self.agents:

            // For each agent get initial observations
            self.obs[agent] = self.aec_env.observe(agent)

        return self.obs

    def step(self, action_dict):

After Change

    def reset(self):
        self.env.reset()
        return {
            self.env.agent_selection: self.env.observe(
                self.env.agent_selection)
        }

    def step(self, action):
        self.env.step(action[self.env.agent_selection])
        obs_d = {}

Instances