Observation renderer throws error
The observation renderer for the "LocalObsForRailEnv" throws error on master branch.
call sequence
initialization
env = RailEnv(size_x,size_y,rail_generator=complex_rail_generato rt = RenderTool(env,gl="PILSVG")
do something with the env
...
use the renderer
rt.render_env(show=True,show_observations=True)
Error Output
Exception has occurred: KeyError 0 File "D:\vt2\try_1\flatland\flatland\utils\rendertools.py", line 276, in render_observation for visited_cell in observation_dict[agent]: File "D:\vt2\try_1\flatland\flatland\utils\rendertools.py", line 567, in render_env_2 self.render_observation(range(env.get_num_agents()), env.dev_obs_dict) File "D:\vt2\try_1\flatland\flatland\utils\rendertools.py", line 419, in render_env selected_agent=selected_agent, action_dict=action_dict) File "D:\vt2\try_1\linker.py", line 383, in collect_training_data env_renderer.render_env(show=True,show_observations=True) File "D:\vt2\try_1\linker.py", line 200, in perform_training predicted_, reward_ ,steps_, done_= self.collect_training_data(env,env_renderer) File "D:\vt2\try_1\main.py", line 70, in link_.perform_training(env,rt)