Skip to content
Snippets Groups Projects
Commit 18a8c10e authored by spiglerg's avatar spiglerg
Browse files

fix?

parent 45480975
No related branches found
No related tags found
No related merge requests found
...@@ -460,7 +460,8 @@ class RailEnv(Environment): ...@@ -460,7 +460,8 @@ class RailEnv(Environment):
def set_full_state_msg(self, msg_data): def set_full_state_msg(self, msg_data):
data = msgpack.unpackb(msg_data, use_list=False) data = msgpack.unpackb(msg_data, use_list=False)
self.rail.grid = np.array(data[b"grid"]) 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"]] self.agents = [EnvAgent(d[0], d[1], d[2], d[3], d[4]) for d in data[b"agents"]]
# setup with loaded data # setup with loaded data
self.height, self.width = self.rail.grid.shape self.height, self.width = self.rail.grid.shape
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment