Skip to content
Snippets Groups Projects
Commit ebbbe6fb authored by u214892's avatar u214892
Browse files

bugfix test_multi_speed

parent 13cce94b
No related branches found
No related tags found
No related merge requests found
...@@ -66,7 +66,7 @@ def test_multi_speed_init(): ...@@ -66,7 +66,7 @@ def test_multi_speed_init():
# Run episode # Run episode
for step in range(100): for step in range(100):
# Chose an action for each agent in the environment # Choose an action for each agent in the environment
for a in range(env.get_num_agents()): for a in range(env.get_num_agents()):
action = agent.act(0) action = agent.act(0)
action_dict.update({a: action}) action_dict.update({a: action})
...@@ -75,12 +75,11 @@ def test_multi_speed_init(): ...@@ -75,12 +75,11 @@ def test_multi_speed_init():
assert old_pos[a] == env.agents[a].position assert old_pos[a] == env.agents[a].position
# Environment step which returns the observations for all agents, their corresponding # Environment step which returns the observations for all agents, their corresponding
# reward and whether their are done # reward and whether they are done
_, _, _, _ = env.step(action_dict) _, _, _, _ = env.step(action_dict)
# Update old position whenever an agent was allowed to move # Update old position whenever an agent was allowed to move
for i_agent in range(env.get_num_agents()): for i_agent in range(env.get_num_agents()):
if (step + 1) % (i_agent + 1) == 0: if (step + 1) % (i_agent + 1) == 0:
print(step, i_agent, env.agents[a].position) print(step, i_agent, env.agents[i_agent].position)
old_pos[i_agent] = env.agents[i_agent].position old_pos[i_agent] = env.agents[i_agent].position
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment