i think i did the converstion correct
This commit is contained in:
@@ -381,12 +381,13 @@ class World:
|
||||
self.agent_composition=agent_composition
|
||||
self.n_agents=0
|
||||
self._agents = []
|
||||
for k,v in agent_composition:
|
||||
for k,v in agent_composition.items():
|
||||
self._agent_class_idx_map[k]=[]
|
||||
for offset in range(v):
|
||||
agent_class=agent_registry.get(k)
|
||||
self._agents.append(agent_class(self.n_agents,multi_action_mode_agents=self.multi_action_mode_agents))
|
||||
self._agent_class_idx_map[k].append(self.n_agents)
|
||||
agent=agent_class(self.n_agents,self.multi_action_mode_agents)
|
||||
self._agents.append(agent)
|
||||
self._agent_class_idx_map[k].append(str(self.n_agents))
|
||||
self.n_agents+=1
|
||||
self.maps = Maps(world_size, self.n_agents, world_resources, world_landmarks)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user