From 67ec235eb51f3ca4e69870c0c598de64c2bdc238 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 20 Jun 2019 08:38:46 +0200
Subject: [PATCH] flake8 conformability
---
examples/simple_example_3.py | 2 +-
tests/test_environments.py | 3 ++-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/examples/simple_example_3.py b/examples/simple_example_3.py
index e015b3c..1661ef6 100644
--- a/examples/simple_example_3.py
+++ b/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
diff --git a/tests/test_environments.py b/tests/test_environments.py
index aa24467..79160ce 100644
--- a/tests/test_environments.py
+++ b/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([])
--
GitLab