From a21b743d83d8b628fa63ee2f0e393d85422457d8 Mon Sep 17 00:00:00 2001
From: Giacomo Spigler <spiglerg@gmail.com>
Date: Wed, 19 Jun 2019 13:06:50 +0200
Subject: [PATCH] test3
---
flatland/envs/agent_utils.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/flatland/envs/agent_utils.py b/flatland/envs/agent_utils.py
index bf6fc0de..d1416766 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):
--
GitLab