Skip to content
Snippets Groups Projects
Commit d3542199 authored by Egli Adrian (IT-SCI-API-PFI)'s avatar Egli Adrian (IT-SCI-API-PFI)
Browse files

realistic generator removed

parent 8bbf5b62
No related branches found
No related tags found
No related merge requests found
This diff is collapsed.
import os from flatland.envs.generators import sparse_rail_generator
import numpy as np
from flatland.envs.generators import sparse_rail_generator, realistic_rail_generator
from flatland.envs.observations import GlobalObsForRailEnv from flatland.envs.observations import GlobalObsForRailEnv
from flatland.envs.rail_env import RailEnv from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool, AgentRenderVariant from flatland.utils.rendertools import RenderTool
def test_realistic_rail_generator(vizualization_folder_name=None):
num_agents = np.random.randint(10, 30)
env = RailEnv(width=np.random.randint(40, 80),
height=np.random.randint(10, 20),
rail_generator=realistic_rail_generator(nr_start_goal=num_agents + 1,
seed=1,
add_max_dead_end=4,
two_track_back_bone=1 % 2 == 0),
number_of_agents=num_agents,
obs_builder_object=GlobalObsForRailEnv())
# reset to initialize agents_static
env_renderer = RenderTool(env, gl="PILSVG", agent_render_variant=AgentRenderVariant.ONE_STEP_BEHIND,
screen_height=600,
screen_width=800)
env_renderer.render_env(show=True, show_observations=True, show_predictions=False)
if vizualization_folder_name is not None:
env_renderer.gl.save_image(
os.path.join(
vizualization_folder_name,
"flatland_frame_{:04d}.png".format(0)
))
env_renderer.close_window()
def test_sparse_rail_generator(): def test_sparse_rail_generator():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment