From bf23a556038ae1c7d56179cd7a62a57b296d103d Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Wed, 10 Jul 2019 15:04:23 +0200
Subject: [PATCH] flake8
---
examples/custom_railmap_example.py | 4 ++--
tests/simple_rail.py | 5 +++--
2 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/examples/custom_railmap_example.py b/examples/custom_railmap_example.py
index 6dc157f3..a76b9c97 100644
--- a/examples/custom_railmap_example.py
+++ b/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 = []
diff --git a/tests/simple_rail.py b/tests/simple_rail.py
index fb739f29..28978ca3 100644
--- a/tests/simple_rail.py
+++ b/tests/simple_rail.py
@@ -1,11 +1,12 @@
-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:
# |
# |
--
GitLab