From 8fb2fb6f363d9c38ae6e1901396337922c00d623 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 1 Nov 2019 16:03:20 -0400
Subject: [PATCH] updated tests
---
tests/test_flatland_envs_predictions.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_flatland_envs_predictions.py b/tests/test_flatland_envs_predictions.py
index 45e0bdda..cfb3e982 100644
--- a/tests/test_flatland_envs_predictions.py
+++ b/tests/test_flatland_envs_predictions.py
@@ -34,7 +34,7 @@ def test_dummy_predictor(rendering=False):
# reset to set agents from agents_static
env.reset(False, False)
- env.set_agent_active(0)
+ env.set_agent_active(env.agents[0])
if rendering:
renderer = RenderTool(env, gl="PILSVG")
--
GitLab