#164 improving stability sparse level generator

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:

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))