From 9bc21050aed7ef222a326e84d0066750c667810f Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 25 Sep 2019 18:33:16 -0400
Subject: [PATCH] fixed bugs in code but got new assertion error in tests.
these need to be investigated before removing
---
tests/test_flatland_malfunction.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_flatland_malfunction.py b/tests/test_flatland_malfunction.py
index 1727dba8..b6e082e3 100644
--- a/tests/test_flatland_malfunction.py
+++ b/tests/test_flatland_malfunction.py
@@ -53,7 +53,7 @@ class SingleAgentNavigationObs(TreeObsForRailEnv):
min_distances.append(np.inf)
observation = [0, 0, 0]
- observation[np.argmin(min_distances)[0]] = 1
+ observation[np.argmin(min_distances)] = 1
return observation
--
GitLab