From 78b0726351c9b910ee76acc0180f5d00cea847bb Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 25 Sep 2019 18:27:31 -0400
Subject: [PATCH] further work on getting test back up
---
tests/test_flatland_malfunction.py | 7 ++-----
1 file changed, 2 insertions(+), 5 deletions(-)
diff --git a/tests/test_flatland_malfunction.py b/tests/test_flatland_malfunction.py
index 1b3c6ade..1727dba8 100644
--- a/tests/test_flatland_malfunction.py
+++ b/tests/test_flatland_malfunction.py
@@ -2,6 +2,7 @@ import random
from typing import Dict
import numpy as np
+from test_utils import Replay, ReplayConfig, run_replay_config, set_penalties_for_replay
from flatland.core.grid.grid4 import Grid4TransitionsEnum
from flatland.core.grid.grid4_utils import get_new_position
@@ -9,7 +10,6 @@ from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.rail_env import RailEnv, RailEnvActions
from flatland.envs.rail_generators import complex_rail_generator, sparse_rail_generator
from flatland.envs.schedule_generators import complex_schedule_generator, sparse_schedule_generator
-from test_utils import Replay, ReplayConfig, run_replay_config, set_penalties_for_replay
class SingleAgentNavigationObs(TreeObsForRailEnv):
@@ -180,8 +180,7 @@ def test_initial_malfunction():
num_neighb=3,
# Number of connections to other cities/intersections
seed=215545, # Random seed
- grid_mode=True,
- enhance_intersection=False
+ grid_mode=True
),
schedule_generator=sparse_schedule_generator(speed_ration_map),
number_of_agents=1,
@@ -264,7 +263,6 @@ def test_initial_malfunction_stop_moving():
# Number of connections to other cities/intersections
seed=215545, # Random seed
grid_mode=True,
- enhance_intersection=False
),
schedule_generator=sparse_schedule_generator(speed_ration_map),
number_of_agents=1,
@@ -357,7 +355,6 @@ def test_initial_malfunction_do_nothing():
# Number of connections to other cities/intersections
seed=215545, # Random seed
grid_mode=True,
- enhance_intersection=False
),
schedule_generator=sparse_schedule_generator(speed_ration_map),
number_of_agents=1,
--
GitLab