From d14d148482ec2b0223a4c137ada3e98e67017846 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Tue, 1 Oct 2019 09:18:44 -0400
Subject: [PATCH] removed debug print
---
examples/flatland_2_0_example.py | 4 ++--
flatland/envs/rail_generators.py | 1 -
2 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index 3ca6ab9d..621f2c79 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -30,14 +30,14 @@ speed_ration_map = {1.: 0.25, # Fast passenger train
env = RailEnv(width=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
grid_mode=False,
max_inter_city_rails=2,
max_tracks_in_city=4,
),
schedule_generator=sparse_schedule_generator(),
- number_of_agents=50,
+ number_of_agents=10,
stochastic_data=stochastic_data, # Malfunction data generator
obs_builder_object=GlobalObsForRailEnv())
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index a1205580..337b301a 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -946,7 +946,6 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4,
if direction_set == 4:
return closest_neighb
- print(closest_neighb)
return closest_neighb
def argsort(seq):
--
GitLab