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

removed debug print

parent fd1651b0
No related branches found
No related tags found
No related merge requests found
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()):
......
......@@ -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)
......
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