i think i did the converstion correct

ai_economist/foundation/base/base_env.py

@@ -221,7 +221,7 @@ class BaseEnvironment(ABC):
         # Set n_agents
         self.agent_composition=agent_composition
         n_agents=0
-        for k,v in agent_composition:
+        for k,v in agent_composition.items():
             n_agents+=v
 
         # Number of agents must be an integer and there must be at least 2 agents
@@ -323,7 +323,7 @@ class BaseEnvironment(ABC):
         # now that we know all the entities we'll use.
         self.world = World(
             self.world_size,
-            self.n_agents,
+            self.agent_composition,
             self.resources,
             self.landmarks,
             self.multi_action_mode_agents,