diff --git a/flatland/envs/agent_utils.py b/flatland/envs/agent_utils.py index bf6fc0de8b72b9996b45e00414ac61c9636cd7ce..d14167669796b031ff7141ab89ddebc16fa01325 100644 --- a/flatland/envs/agent_utils.py +++ b/flatland/envs/agent_utils.py @@ -29,9 +29,9 @@ class EnvAgentStatic(object): direction = attrib() target = attrib() moving = attrib() - speed_data = attrib() + speed_data = attrib(default=dict({'position_fraction':0.0, 'speed':1.0, 'transition_action_on_cellexit':0})) - def __init__(self, position, direction, target, moving=False, speed_data={'position_fraction':0.0, 'speed':1.0, 'transition_action_on_cellexit':2}): + def __init__(self, position, direction, target, moving=False, speed_data={'position_fraction':0.0, 'speed':1.0, 'transition_action_on_cellexit':0}): self.position = position self.direction = direction self.target = target @@ -44,7 +44,7 @@ class EnvAgentStatic(object): """ speed_datas = [] for i in range(len(positions)): - speed_datas.append( {'position_fraction':0.0, 'speed':1.0, 'transition_action_on_cellexit':2} ) + speed_datas.append( {'position_fraction':0.0, 'speed':1.0, 'transition_action_on_cellexit':0} ) return list(starmap(EnvAgentStatic, zip(positions, directions, targets, [False] * len(positions), speed_datas))) def to_list(self):