From 18a8c10ee7623d2a12801eaf97df7b38606642ba Mon Sep 17 00:00:00 2001 From: Giacomo Spigler <spiglerg@gmail.com> Date: Wed, 5 Jun 2019 22:32:27 +0200 Subject: [PATCH] fix? --- flatland/envs/rail_env.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index b9373770..77ac6c06 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -460,7 +460,8 @@ class RailEnv(Environment): def set_full_state_msg(self, msg_data): data = msgpack.unpackb(msg_data, use_list=False) self.rail.grid = np.array(data[b"grid"]) - self.agents_static = [EnvAgentStatic(d[0], d[1], d[2], d[3]) for d in data[b"agents_static"]] + # agents are always reset as not moving + self.agents_static = [EnvAgentStatic(d[0], d[1], d[2], moving=False) for d in data[b"agents_static"]] self.agents = [EnvAgent(d[0], d[1], d[2], d[3], d[4]) for d in data[b"agents"]] # setup with loaded data self.height, self.width = self.rail.grid.shape -- GitLab