From 5e2701314d0f7dd6b3100d703580092707b8b39d Mon Sep 17 00:00:00 2001 From: MLErik <baerenjesus@gmail.com> Date: Mon, 28 Oct 2019 15:19:35 -0400 Subject: [PATCH] this adresses #263 --- flatland/envs/agent_utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/flatland/envs/agent_utils.py b/flatland/envs/agent_utils.py index 975bfe70..78ec65a3 100644 --- a/flatland/envs/agent_utils.py +++ b/flatland/envs/agent_utils.py @@ -62,10 +62,11 @@ class EnvAgentStatic(object): malfunction_datas = [] for i in range(len(schedule.agent_positions)): malfunction_datas.append({'malfunction': 0, - 'malfunction_rate': schedule.agent_malfunction_rates[i] if schedule.agent_malfunction_rates is not None else 0., + 'malfunction_rate': schedule.agent_malfunction_rates[ + i] if schedule.agent_malfunction_rates is not None else 0., 'next_malfunction': 0, 'nr_malfunctions': 0, - 'fixed':False}) + 'fixed': False}) return list(starmap(EnvAgentStatic, zip(schedule.agent_positions, schedule.agent_directions, -- GitLab