Skip to content

Commit

Permalink
Merge pull request #46 from JoseLuisC99/new-pettingzoo-api-call
Browse files Browse the repository at this point in the history
Fixed broken restart compatibility with the PettingZoo API
  • Loading branch information
atstarke authored Sep 24, 2024
2 parents 1053931 + 9706565 commit ae650fb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion magent2/environments/magent_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,7 +165,7 @@ def reset(self, seed=None, return_info=False, options=None):
self.frames = 0
self.team_sizes = [self.env.get_num(handle) for handle in self.handles]
self.generate_map()
return self._compute_observations()
return self._compute_observations(), {agent: {} for agent in self.agents}

def _compute_observations(self):
observes = [None] * self.max_num_agents
Expand Down

0 comments on commit ae650fb

Please sign in to comment.