From 7fe36fcadaa6dcedde9538a6988541c83cbaeac3 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Thu, 3 Oct 2019 16:29:19 -0400
Subject: [PATCH] minor updates
---
tests/test_global_observation.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/tests/test_global_observation.py b/tests/test_global_observation.py
index fe3e9ec1..9035062b 100644
--- a/tests/test_global_observation.py
+++ b/tests/test_global_observation.py
@@ -23,15 +23,14 @@ def test_get_global_observation():
env = RailEnv(width=50,
height=50,
- rail_generator=sparse_rail_generator(max_num_cities=25,
+ rail_generator=sparse_rail_generator(max_num_cities=6,
max_rails_between_cities=4,
seed=15,
- grid_mode=True
+ grid_mode=False
),
schedule_generator=sparse_schedule_generator(speed_ration_map),
number_of_agents=number_of_agents, stochastic_data=stochastic_data, # Malfunction data generator
obs_builder_object=GlobalObsForRailEnv())
-
obs, all_rewards, done, _ = env.step({0: 0})
for i in range(len(env.agents)):
--
GitLab