From ae4c9178e3fba080cb71e64808204defd5b18146 Mon Sep 17 00:00:00 2001
From: Guillaume Mollard <guillaume.mollard2@gmail.com>
Date: Fri, 24 May 2019 15:13:32 +0200
Subject: [PATCH] fixed test fail for Global Obs
---
tests/test_env_observation_builder.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_env_observation_builder.py b/tests/test_env_observation_builder.py
index 13f8d8f..4038fe6 100644
--- a/tests/test_env_observation_builder.py
+++ b/tests/test_env_observation_builder.py
@@ -82,7 +82,7 @@ def test_global_obs():
# If this assertion is wrong, it means that the observation returned
# places the agent on an empty cell
- assert(np.sum(rail_map * global_obs[0][1][0]) > 0)
+ assert(np.sum(rail_map * global_obs[0][1][:, :, 0]) > 0)
def main():
--
GitLab