fixed bugs in code but got new assertion error in tests. these need to be...

fixed bugs in code but got new assertion error in tests. these need to be investigated before removing

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