From efc4c6b31202f51c2ac3a538e25f16a923baf03e Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Thu, 5 Sep 2019 15:59:10 +0200
Subject: [PATCH] #164 improving stability sparse level generator
---
flatland/envs/rail_env.py | 2 --
flatland/envs/rail_generators.py | 1 +
2 files changed, 1 insertion(+), 2 deletions(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index cc115c72..ddd8e442 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -395,8 +395,6 @@ class RailEnv(Environment):
self.rewards_dict[i_agent] += self.step_penalty * agent.speed_data['speed']
self.rewards_dict[i_agent] += self.stop_penalty
agent.moving = False
- else:
- agent.speed_data['transition_action_on_cellexit'] = action
# if we've just broken in this step, nothing else to do
if new_malfunction:
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 7515009c..5af4a079 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -682,6 +682,7 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
if tries > 100:
warnings.warn("Could not set trainstations, please change initial parameters!!!!")
spot_found = False
+ break
if spot_found:
train_stations[trainstation_node].append((station_x, station_y))
--
GitLab