From 1868a59a5575135c9b8c4488cf2978832e29cb25 Mon Sep 17 00:00:00 2001 From: Giacomo Spigler <spiglerg@gmail.com> Date: Tue, 6 Aug 2019 15:53:15 +0200 Subject: [PATCH] flake8 --- flatland/envs/agent_utils.py | 10 ++++++++-- flatland/envs/rail_env.py | 2 +- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/flatland/envs/agent_utils.py b/flatland/envs/agent_utils.py index 6b881f9c..2017d706 100644 --- a/flatland/envs/agent_utils.py +++ b/flatland/envs/agent_utils.py @@ -38,13 +38,19 @@ class EnvAgentStatic(object): 'speed': speeds[i] if speeds is not None else 1.0, 'transition_action_on_cellexit': 0}) - # TODO: on initialization, all agents are re-set as non-broken. Perhaps it may be desirable to set some as broken? + # TODO: on initialization, all agents are re-set as non-broken. Perhaps it may be desirable to set + # some as broken? broken_datas = [] for i in range(len(positions)): broken_datas.append({'broken': 0, 'number_of_halts': 0}) - return list(starmap(EnvAgentStatic, zip(positions, directions, targets, [False] * len(positions), speed_datas, broken_datas))) + return list(starmap(EnvAgentStatic, zip(positions, + directions, + targets, + [False] * len(positions), + speed_datas, + broken_datas))) def to_list(self): diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index 79c86955..0f03870b 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -167,7 +167,7 @@ class RailEnv(Environment): # Stochastic train breaking parameters self.min_average_broken_trains = 1 - self.average_proportion_of_broken_trains = 0.1 # ~10% of the trains can be expected to break down in an episode + self.average_proportion_of_broken_trains = 0.1 # ~10% of the trains can be expected to break down in an episode self.mean_number_halts_to_break = 3 # Uniform distribution -- GitLab