diff --git a/flatland/envs/agent_utils.py b/flatland/envs/agent_utils.py index 975bfe708b4e05eb17714c28d5117c6417526c84..78ec65a3c65f0a4dad72071d97e173fd636bba90 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,