Skip to content
Snippets Groups Projects
Commit 44b0c053 authored by spmohanty's avatar spmohanty
Browse files

Remove explicit references to render mode PILSVG, etc

parent c65ccbe9
No related branches found
No related tags found
No related merge requests found
...@@ -147,7 +147,7 @@ Basic usage of the RailEnv environment used by the Flatland Challenge ...@@ -147,7 +147,7 @@ Basic usage of the RailEnv environment used by the Flatland Challenge
seed=0), seed=0),
number_of_agents=2) number_of_agents=2)
env_renderer = RenderTool(env, gl="PILSVG") env_renderer = RenderTool(env)
for step in range(100): for step in range(100):
obs, all_rewards, done, _ = env.step( obs, all_rewards, done, _ = env.step(
......
...@@ -75,7 +75,7 @@ class Demo: ...@@ -75,7 +75,7 @@ class Demo:
self.record_frames = record_frames self.record_frames = record_frames
def create_renderer(self): def create_renderer(self):
self.renderer = RenderTool(self.env, gl="PILSVG") self.renderer = RenderTool(self.env)
handle = self.env.get_agent_handles() handle = self.env.get_agent_handles()
return handle return handle
......
...@@ -93,7 +93,7 @@ def max_lt(seq, val): ...@@ -93,7 +93,7 @@ def max_lt(seq, val):
return None return None
def main(render=True, delay=0.0, n_trials=3, n_steps=50, sGL="PILSVG"): def main(render=True, delay=0.0, n_trials=3, n_steps=50):
random.seed(1) random.seed(1)
np.random.seed(1) np.random.seed(1)
...@@ -103,7 +103,7 @@ def main(render=True, delay=0.0, n_trials=3, n_steps=50, sGL="PILSVG"): ...@@ -103,7 +103,7 @@ def main(render=True, delay=0.0, n_trials=3, n_steps=50, sGL="PILSVG"):
number_of_agents=5) number_of_agents=5)
if render: if render:
env_renderer = RenderTool(env, gl=sGL) env_renderer = RenderTool(env)
oPlayer = Player(env) oPlayer = Player(env)
......
...@@ -18,7 +18,7 @@ env = RailEnv(width=6, ...@@ -18,7 +18,7 @@ env = RailEnv(width=6,
env.reset() env.reset()
env_renderer = RenderTool(env, gl="PILSVG") env_renderer = RenderTool(env)
env_renderer.renderEnv(show=True) env_renderer.renderEnv(show=True)
env_renderer.renderEnv(show=True) env_renderer.renderEnv(show=True)
......
...@@ -21,7 +21,7 @@ obs, all_rewards, done, _ = env.step({0: 0}) ...@@ -21,7 +21,7 @@ obs, all_rewards, done, _ = env.step({0: 0})
for i in range(env.get_num_agents()): for i in range(env.get_num_agents()):
env.obs_builder.util_print_obs_subtree(tree=obs[i], num_features_per_node=7) env.obs_builder.util_print_obs_subtree(tree=obs[i], num_features_per_node=7)
env_renderer = RenderTool(env, gl="PILSVG") env_renderer = RenderTool(env)
env_renderer.renderEnv(show=True, frames=True) env_renderer.renderEnv(show=True, frames=True)
env_renderer.renderEnv(show=True, frames=True) env_renderer.renderEnv(show=True, frames=True)
......
from examples.play_model import Player try:
from examples.play_model import Player
except ImportError:
from play_model import Player
from flatland.envs.generators import complex_rail_generator from flatland.envs.generators import complex_rail_generator
from flatland.envs.rail_env import RailEnv from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool from flatland.utils.rendertools import RenderTool
......
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