flake8 conformability

examples/simple_example_3.py

@@ -2,7 +2,7 @@ import random
 
 import numpy as np
 
-from flatland.envs.generators import random_rail_generator, complex_rail_generator
+from flatland.envs.generators import random_rail_generator
 from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool

tests/test_environments.py

@@ -107,7 +107,6 @@ def test_rail_environment_single_agent():
             if prev_pos != pos:
                 valid_active_actions_done += 1
 
-
         # After 6 movements on this railway network, the train should be back
         # to its original height on the map.
         assert (initial_pos[0] == agent.position[0])
@@ -124,8 +123,10 @@ def test_rail_environment_single_agent():
                 _, _, dones, _ = rail_env.step({0: action})
                 done = dones['__all__']
 
+
 test_rail_environment_single_agent()
 
+
 def test_dead_end():
     transitions = Grid4Transitions([])