diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py index 161927a49c2dcb71aef20c369e9293ca2b24eb57..0099b1cd88c2939ea3c4a04b308792a1403d716b 100644 --- a/examples/flatland_2_0_example.py +++ b/examples/flatland_2_0_example.py @@ -1,5 +1,3 @@ -import time - import numpy as np from flatland.envs.observations import TreeObsForRailEnv, GlobalObsForRailEnv @@ -115,7 +113,6 @@ for step in range(500): # reward and whether their are done next_obs, all_rewards, done, _ = env.step(action_dict) env_renderer.render_env(show=True, show_observations=False, show_predictions=False) - time.sleep(10) frame_step += 1 # Update replay buffer and train agent for a in range(env.get_num_agents()): diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py index e1128fc7d0b07c1759920d430ec3547750193d73..226b7f891f8ca182b23930e66fe828a0fac0fa0f 100644 --- a/flatland/envs/rail_generators.py +++ b/flatland/envs/rail_generators.py @@ -581,8 +581,7 @@ def sparse_rail_generator(num_cities=5, min_node_dist=20, node_radius=2, inner_connection_points, outer_connection_points, connection_info = _generate_node_connection_points( node_positions, node_radius, max_inter_city_rails_allowed, tracks_in_city) - print(inner_connection_points) - print(outer_connection_points) + # Connect the cities through the connection points _connect_cities(node_positions, outer_connection_points, connection_info, city_cells, rail_trans, grid_map)