Jupyter notebooks are ready for flatland3.

But they should be overworked. Might just one example per use case would be much more clear. TODO - cleanup

Jupyter notebooks are ready for flatland3.
But they should be overworked. Might just one example per use case would be much more clear. TODO - cleanup
f4bc62a8 · adrian_egli2 · 009d2b0e · f4bc62a8
Commit f4bc62a8 authored 2 years ago by adrian_egli2
--- a/notebooks/flatland_animate.ipynb
+++ b/notebooks/flatland_animate.ipynb
@@ -48,8 +48,8 @@
   "metadata": {},
   "outputs": [],
   "source": [
-    "oRT = RenderTool(env)\n",
+    "oRT = RenderTool(env, gl=\"PILSVG\")\n",
-    "oRT.render_env()"
+    "oRT.render_env(show_observations=False,show_predictions=False)"
   ]
  },
  {
@@ -71,7 +71,7 @@
    "    for agent_handle, agents in enumerate(env.agents):\n",
    "        actions.update({agent_handle:rea.MOVE_FORWARD})\n",
    "    obs, rew, done, info = env.step(actions)\n",
-    "    oRT.render_env()\n",
+    "    oRT.render_env(show_observations=False,show_predictions=False)\n",
    "    gIm = oRT.get_image()\n",
    "    canvas.put_image_data(gIm)\n",
    "    time.sleep(0.1)"

 %% Cell type:markdown id: tags:
 # Simple Animation Demo
 %% Cell type:code id: tags:
 ``` python
 %load_ext autoreload
 %autoreload 2
 ```
 %% Cell type:code id: tags:
 ``` python
 from flatland.utils.rendertools import RenderTool
 import ipycanvas
 import time
 from flatland.envs.rail_env import RailEnv
 from flatland.envs.rail_env import RailEnvActions as rea
 from flatland.envs.persistence import RailEnvPersister
 ```
 %% Cell type:code id: tags:
 ``` python
 env, env_dict = RailEnvPersister.load_new("complex_scene_2.pkl", load_from_package="env_data.railway")
 _ = env.reset()
 env._max_episode_steps = 100
 ```
 %% Cell type:code id: tags:
 ``` python
-oRT = RenderTool(env)
+oRT = RenderTool(env, gl="PILSVG")
-oRT.render_env()
+oRT.render_env(show_observations=False,show_predictions=False)
 ```
 %% Cell type:code id: tags:
 ``` python
 image_arr = oRT.get_image()
 canvas = ipycanvas.Canvas()
 canvas.put_image_data(image_arr)
 display(canvas)
 done={"__all__":False}
 while not done["__all__"]:
    actions = {}
    for agent_handle, agents in enumerate(env.agents):
        actions.update({agent_handle:rea.MOVE_FORWARD})
    obs, rew, done, info = env.step(actions)
-    oRT.render_env()
+    oRT.render_env(show_observations=False,show_predictions=False)
    gIm = oRT.get_image()
    canvas.put_image_data(gIm)
    time.sleep(0.1)
 ```