From 35cc3bc429fd7008c94e05359ed6c5d5660e0b30 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 27 Sep 2019 18:59:29 -0400
Subject: [PATCH] removed debug print
---
examples/flatland_2_0_example.py | 3 ---
flatland/envs/rail_generators.py | 3 +--
2 files changed, 1 insertion(+), 5 deletions(-)
diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index 161927a4..0099b1cd 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 e1128fc7..226b7f89 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)
--
GitLab