From e853ad7c3a8a5d994cea3d50d70fdadeee31cb04 Mon Sep 17 00:00:00 2001 From: u214892 <u214892@sbb.ch> Date: Wed, 10 Jul 2019 16:17:55 +0200 Subject: [PATCH] fix flake8 --- examples/custom_railmap_example.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/custom_railmap_example.py b/examples/custom_railmap_example.py index 6dc157f..a76b9c9 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 = [] -- GitLab