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

removed debug print

parent d6a45659
No related branches found
No related tags found
No related merge requests found
...@@ -30,14 +30,14 @@ speed_ration_map = {1.: 0.25, # Fast passenger train ...@@ -30,14 +30,14 @@ speed_ration_map = {1.: 0.25, # Fast passenger train
env = RailEnv(width=50, env = RailEnv(width=50,
height=50, height=50,
rail_generator=sparse_rail_generator(num_cities=2, # Number of cities in map (where train stations are) rail_generator=sparse_rail_generator(num_cities=8, # Number of cities in map (where train stations are)
seed=1, # Random seed seed=1, # Random seed
grid_mode=False, grid_mode=False,
max_inter_city_rails=2, max_inter_city_rails=2,
max_tracks_in_city=4, max_tracks_in_city=4,
), ),
schedule_generator=sparse_schedule_generator(), schedule_generator=sparse_schedule_generator(),
number_of_agents=50, number_of_agents=10,
stochastic_data=stochastic_data, # Malfunction data generator stochastic_data=stochastic_data, # Malfunction data generator
obs_builder_object=GlobalObsForRailEnv()) obs_builder_object=GlobalObsForRailEnv())
......
...@@ -946,7 +946,6 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4, ...@@ -946,7 +946,6 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4,
if direction_set == 4: if direction_set == 4:
return closest_neighb return closest_neighb
print(closest_neighb)
return closest_neighb return closest_neighb
def argsort(seq): def argsort(seq):
......
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