From 22bf52f36b730369b2e98e2335ef809919059d3a Mon Sep 17 00:00:00 2001 From: Erik Nygren <erik.nygren@sbb.ch> Date: Sun, 14 Jul 2019 13:27:19 -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 fixed typos --- flatland/envs/observations.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py index 15d42d7..2f8ffb8 100644 --- a/flatland/envs/observations.py +++ b/flatland/envs/observations.py @@ -22,8 +22,6 @@ class TreeObsForRailEnv(ObservationBuilder): For details about the features in the tree observation see the get() function. """ - - def __init__(self, max_depth, predictor=None): super().__init__() self.max_depth = max_depth -- GitLab