From 842be3959bb48b02a5d0fc557d50e8d2e94a705a Mon Sep 17 00:00:00 2001
From: u229589 <christian.baumberger@sbb.ch>
Date: Mon, 16 Sep 2019 19:52:35 +0200
Subject: [PATCH] Refactoring: prediction_builder knows its environment and can
access the distance map directly
---
docs/intro_observationbuilder.rst | 2 +-
flatland/envs/observations.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/docs/intro_observationbuilder.rst b/docs/intro_observationbuilder.rst
index 563dd113..64e953da 100644
--- a/docs/intro_observationbuilder.rst
+++ b/docs/intro_observationbuilder.rst
@@ -205,7 +205,7 @@ In contrast to the previous examples we also implement the :code:`def get_many(s
:return:
'''
- self.predictions = self.predictor.get(custom_args={'distance_map': self.env.distance_map})
+ self.predictions = self.predictor.get()
self.predicted_pos = {}
for t in range(len(self.predictions[0])):
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index baa378f1..3f398c74 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -71,7 +71,7 @@ class TreeObsForRailEnv(ObservationBuilder):
self.max_prediction_depth = 0
self.predicted_pos = {}
self.predicted_dir = {}
- self.predictions = self.predictor.get(custom_args={'distance_map': self.env.distance_map})
+ self.predictions = self.predictor.get()
if self.predictions:
for t in range(len(self.predictions[0])):
--
GitLab