From a98dafa546059718a28d51f909f733511d9c914f Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sun, 14 Jul 2019 13:26:55 -0400
Subject: [PATCH] updated load and save function. Now also distance maps are
stored. Additional package msgpack-numpy needed for ndarray. This saves tons
of time when loading precomputed files. Updated test to test loading and
saving with and without distance maps minor update to bug in observation
dimension
---
flatland/envs/observations.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 4385d4da..15d42d79 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -22,7 +22,7 @@ class TreeObsForRailEnv(ObservationBuilder):
For details about the features in the tree observation see the get() function.
"""
- observation_dim = 9
+
def __init__(self, max_depth, predictor=None):
super().__init__()
@@ -34,6 +34,7 @@ class TreeObsForRailEnv(ObservationBuilder):
for i in range(self.max_depth + 1):
size += pow4
pow4 *= 4
+ self.observation_dim = 9
self.observation_space = [size * self.observation_dim]
self.location_has_agent = {}
self.location_has_agent_direction = {}
--
GitLab