test3

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):