fixed bug in connection

examples/flatland_2_0_example.py

@@ -30,7 +30,7 @@ speed_ration_map = {1.: 0.25,  # Fast passenger train
 
 env = RailEnv(width=50,
               height=50,
-              rail_generator=sparse_rail_generator(num_cities=10,  # Number of cities in map (where train stations are)
+              rail_generator=sparse_rail_generator(num_cities=2,  # Number of cities in map (where train stations are)
                                                    seed=1,  # Random seed
                                                    grid_mode=False,
                                                    max_inter_city_rails=2,

flatland/envs/rail_generators.py

@@ -753,7 +753,7 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4,
                         tmp_direction = (out_direction - 1) % 4
                     while neighb_idx is None:
                         neighb_idx = neighbours[tmp_direction]
-                        tmp_direction = (out_direction + 1) % 4
+                        tmp_direction = (tmp_direction + 1) % 4
                     min_connection_dist = np.inf
                     for dir in range(4):
                         current_points = connection_points[neighb_idx][dir]
@@ -946,7 +946,7 @@ 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):