From 235652ab33f26684e23eeacf1fbb7bdabffaab0a Mon Sep 17 00:00:00 2001
From: spiglerg <spiglerg@gmail.com>
Date: Mon, 20 May 2019 14:02:21 +0200
Subject: [PATCH] fixed issue #18
---
flatland/envs/observations.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 651d835..fa77b58 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -492,8 +492,11 @@ class GlobalObsForRailEnv(ObservationBuilder):
self.rail_obs = np.zeros((self.env.height, self.env.width, 16))
for i in range(self.rail_obs.shape[0]):
for j in range(self.rail_obs.shape[1]):
- self.rail_obs[i, j] = np.array(
- list(f'{self.env.rail.get_transitions((i, j)):016b}')).astype(int)
+ bitlist = [int(digit) for digit in bin(self.env.rail.get_transitions((i, j)))[2:]]
+ bitlist = [0] * (16 - len(bitlist)) + bitlist
+ self.rail_obs[i, j] = np.array(bitlist)
+ # self.rail_obs[i, j] = np.array(
+ # list(f'{self.env.rail.get_transitions((i, j)):016b}')).astype(int)
# self.targets = np.zeros(self.env.height, self.env.width)
# for target_pos in self.env.agents_target:
--
GitLab