Skip to content
Snippets Groups Projects
Commit ce2e149d authored by Erik Nygren's avatar Erik Nygren
Browse files

updated example for flatland 2.0

parent 5efa00dc
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
examples/flatland_2_0_example.py
+ 0
− 3
View file @ ce2e149d
import time
import numpy as np import numpy as np
from flatland.envs.observations import TreeObsForRailEnv from flatland.envs.observations import TreeObsForRailEnv
...@@ -114,7 +112,6 @@ for step in range(500): ...@@ -114,7 +112,6 @@ for step in range(500):
next_obs, all_rewards, done, _ = env.step(action_dict) next_obs, all_rewards, done, _ = env.step(action_dict)
env_renderer.render_env(show=True, show_observations=False, show_predictions=False) env_renderer.render_env(show=True, show_observations=False, show_predictions=False)
frame_step += 1 frame_step += 1
time.sleep(10.1)
# Update replay buffer and train agent # Update replay buffer and train agent
for a in range(env.get_num_agents()): for a in range(env.get_num_agents()):
agent.step((obs[a], action_dict[a], all_rewards[a], next_obs[a], done[a])) agent.step((obs[a], action_dict[a], all_rewards[a], next_obs[a], done[a]))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment