flake8

examples/custom_railmap_example.py

@@ -2,8 +2,8 @@ import random
 
 import numpy as np
 
-from flatland.core.transition_map import GridTransitionMap
 from flatland.core.grid.rail_env_grid import RailEnvTransitions
+from flatland.core.transition_map import GridTransitionMap
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool
 
@@ -17,7 +17,7 @@ def custom_rail_generator():
         grid_map = GridTransitionMap(width=width, height=height, transitions=rail_trans)
         rail_array = grid_map.grid
         rail_array.fill(0)
-        new_tran = rail_trans.set_transition(1,1,1,1)
+        new_tran = rail_trans.set_transition(1, 1, 1, 1)
         print(new_tran)
         agents_positions = []
         agents_direction = []

tests/simple_rail.py

-import numpy as np
 from typing import Tuple
 
+import numpy as np
+
 from flatland.core.grid.grid4 import Grid4Transitions
 from flatland.core.transition_map import GridTransitionMap
 
 
-def make_simple_rail() -> Tuple[GridTransitionMap,np.array]:
+def make_simple_rail() -> Tuple[GridTransitionMap, np.array]:
     # We instantiate a very simple rail network on a 7x10 grid:
     #        |
     #        |