From 302fb144afa2b70a32766f66041a5d20424451ad Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Wed, 12 Jun 2019 16:10:08 +0200
Subject: [PATCH] #57 flake8 cleanup
---
examples/demo.py | 4 +++-
flatland/envs/generators.py | 2 +-
flatland/utils/graphics_pil.py | 1 -
3 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/examples/demo.py b/examples/demo.py
index ae2c872..fb7196b 100644
--- a/examples/demo.py
+++ b/examples/demo.py
@@ -165,4 +165,6 @@ class Demo:
demo_001.set_record_frames(os.path.join(__file_dirname__, '..', 'rendering', 'frame_{:04d}.bmp'))
demo_001.run_demo(360)
-Demo.run_complex_scene()
+
+if __name__ == "__main__":
+ Demo.run_complex_scene()
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 791a07a..f644bc1 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -1,7 +1,7 @@
import numpy as np
from flatland.core.transition_map import GridTransitionMap
-from flatland.core.transitions import Grid8Transitions, RailEnvTransitions
+from flatland.core.transitions import RailEnvTransitions
from flatland.envs.env_utils import distance_on_rail, connect_rail, get_direction, mirror
from flatland.envs.env_utils import get_rnd_agents_pos_tgt_dir_on_rail
diff --git a/flatland/utils/graphics_pil.py b/flatland/utils/graphics_pil.py
index e6a92d3..be804c0 100644
--- a/flatland/utils/graphics_pil.py
+++ b/flatland/utils/graphics_pil.py
@@ -384,7 +384,6 @@ class PILSVG(PILGL):
}
# "paint" color of the train images we load
- # a3BaseColor = self.rgb_s2i("0091ea")
a3BaseColor = self.rgb_s2i("d50000")
self.dPilZug = {}
--
GitLab