From c05ca5e61308ddf7704113cd1ca15e7ea6fa9dab Mon Sep 17 00:00:00 2001
From: u229589 <christian.baumberger@sbb.ch>
Date: Mon, 30 Sep 2019 09:45:59 +0200
Subject: [PATCH] make tree_explorted_actions_char as static class variable
---
flatland/envs/observations.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 95e5aeaf..2187c598 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -28,6 +28,9 @@ class TreeObsForRailEnv(ObservationBuilder):
'num_agents_malfunctioning '
'speed_min_fractional '
'childs')
+
+ tree_explorted_actions_char = ['L', 'F', 'R', 'B']
+
"""
TreeObsForRailEnv object.
@@ -46,7 +49,6 @@ class TreeObsForRailEnv(ObservationBuilder):
self.location_has_agent_direction = {}
self.predictor = predictor
self.location_has_target = None
- self.tree_explorted_actions_char = ['L', 'F', 'R', 'B']
def reset(self):
self.location_has_target = {tuple(agent.target): 1 for agent in self.env.agents}
--
GitLab