diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index b93737709abf97e1364c389665431c375b2c16ac..77ac6c06f6baaf8cce29444eda554ff9fca19842 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