[fix] Fix infos in env.reset()

smartgrid/environment.py

@@ -260,7 +260,10 @@ def reset(
         self.reward_calculator.reset()
 
         obs = self._get_obs()
-        return obs, {}
+        # PettingZoo requires the infos to contain a dictionary for each agent,
+        # even if the dictionary itself is empty.
+        infos = {agent_name: {} for agent_name in self.agents}
+        return obs, infos
 
     def render(self, mode='text'):
         """
@@ -333,10 +336,10 @@ def _get_info(self, rewards: RewardsDict) -> InfoDict:
             only the agent's reward, indexed by ``'reward'``.
         """
         return {
-            agent.name: {
-                'reward': rewards[agent.name]
+            agent_name: {
+                'reward': rewards[agent_name]
             }
-            for agent in self.world.agents
+            for agent_name in self.agents
         }
 
     @property