Skip to content
Snippets Groups Projects
Commit 7fe36fca authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

minor updates

parent cfca5902
No related branches found
No related tags found
No related merge requests found
...@@ -23,15 +23,14 @@ def test_get_global_observation(): ...@@ -23,15 +23,14 @@ def test_get_global_observation():
env = RailEnv(width=50, env = RailEnv(width=50,
height=50, height=50,
rail_generator=sparse_rail_generator(max_num_cities=25, rail_generator=sparse_rail_generator(max_num_cities=6,
max_rails_between_cities=4, max_rails_between_cities=4,
seed=15, seed=15,
grid_mode=True grid_mode=False
), ),
schedule_generator=sparse_schedule_generator(speed_ration_map), schedule_generator=sparse_schedule_generator(speed_ration_map),
number_of_agents=number_of_agents, stochastic_data=stochastic_data, # Malfunction data generator number_of_agents=number_of_agents, stochastic_data=stochastic_data, # Malfunction data generator
obs_builder_object=GlobalObsForRailEnv()) obs_builder_object=GlobalObsForRailEnv())
obs, all_rewards, done, _ = env.step({0: 0}) obs, all_rewards, done, _ = env.step({0: 0})
for i in range(len(env.agents)): for i in range(len(env.agents)):
......
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