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
import numpy as np
from flatland.envs.generators import sparse_rail_generator, realistic_rail_generator
from flatland.envs.generators import sparse_rail_generator
from flatland.envs.observations import GlobalObsForRailEnv
from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool, AgentRenderVariant
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()
from flatland.utils.rendertools import RenderTool
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