From 2671b903c2caae9e16f09430093c68b8db07f6c4 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Thu, 13 Jun 2019 09:10:17 +0200
Subject: [PATCH] fix test_env_prediction_builder
---
flatland/core/env.py | 1 -
flatland/envs/observations.py | 1 -
tests/test_env_prediction_builder.py | 2 +-
3 files changed, 1 insertion(+), 3 deletions(-)
diff --git a/flatland/core/env.py b/flatland/core/env.py
index d1f814f..1bc5b6f 100644
--- a/flatland/core/env.py
+++ b/flatland/core/env.py
@@ -84,7 +84,6 @@ class Environment:
"""
raise NotImplementedError()
-
def get_agent_handles(self):
"""
Returns a list of agents' handles to be used as keys in the step()
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 3100e8b..541f8ad 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -295,7 +295,6 @@ class TreeObsForRailEnv(ObservationBuilder):
other_target_encountered = np.inf
other_agent_same_direction = 0
other_agent_opposite_direction = 0
- possible_conflict = 0
num_steps = 1
while exploring:
diff --git a/tests/test_env_prediction_builder.py b/tests/test_env_prediction_builder.py
index be065d3..a1c951d 100644
--- a/tests/test_env_prediction_builder.py
+++ b/tests/test_env_prediction_builder.py
@@ -64,7 +64,7 @@ def test_predictions():
height=rail_map.shape[0],
rail_generator=rail_from_GridTransitionMap_generator(rail),
number_of_agents=1,
- obs_builder_object=TreeObsForRailEnv(max_depth=20, predictor=DummyPredictorForRailEnv(max_depth=20)),
+ obs_builder_object=TreeObsForRailEnv(max_depth=2, predictor=DummyPredictorForRailEnv(max_depth=20)),
)
env.reset()
--
GitLab