Ready for Flatland3

35360bf7 · adrian_egli2 · fa65d360 · 35360bf7
Commit 35360bf7 authored 2 years ago by adrian_egli2
--- a/notebooks/flatland-animate.ipynb
+++ b/notebooks/flatland-animate.ipynb
@@ -9,7 +9,7 @@
  },
  {
   "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
@@ -19,7 +19,7 @@
  },
  {
   "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
@@ -33,44 +33,56 @@
  },
  {
   "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
   "metadata": {},
-   "outputs": [
+   "outputs": [],
-    {
+   "source": [
-     "ename": "AttributeError",
+    "env, env_dict = RailEnvPersister.load_new(\"complex_scene_2.pkl\", load_from_package=\"env_data.railway\")\n",
-     "evalue": "Can't get attribute 'RailAgentStatus' on <module 'flatland.envs.agent_utils' from 'c:\\\\users\\\\u216993\\\\.conda\\\\envs\\\\flatland3-rl\\\\lib\\\\site-packages\\\\flatland\\\\envs\\\\agent_utils.py'>",
+    "_ = env.reset()\n",
-     "output_type": "error",
+    "env._max_episode_steps = 100"
-     "traceback": [
+   ]
-      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
+  },
-      "\u001b[1;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
+  {
-      "\u001b[1;32m<ipython-input-3-24ea30f53e36>\u001b[0m in \u001b[0;36m<module>\u001b[1;34m\u001b[0m\n\u001b[1;32m----> 1\u001b[1;33m \u001b[0menv\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0menv_dict\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mRailEnvPersister\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mload_new\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"test_env_loop.pkl\"\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mload_from_package\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;34m\"env_data.tests\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m\u001b[0;32m      2\u001b[0m \u001b[0moRT\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mRenderTool\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0menv\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m      3\u001b[0m \u001b[0moRT\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mrender_env\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m      4\u001b[0m \u001b[0mimage_arr\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0moRT\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mget_image\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m      5\u001b[0m \u001b[0mcanvas\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mipycanvas\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mCanvas\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
+   "cell_type": "code",
-      "\u001b[1;32mc:\\users\\u216993\\.conda\\envs\\flatland3-rl\\lib\\site-packages\\flatland\\envs\\persistence.py\u001b[0m in \u001b[0;36mload_new\u001b[1;34m(cls, filename, load_from_package)\u001b[0m\n\u001b[0;32m    105\u001b[0m     \u001b[1;32mdef\u001b[0m \u001b[0mload_new\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mcls\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mfilename\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mload_from_package\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;32mNone\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m    106\u001b[0m \u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m--> 107\u001b[1;33m         \u001b[0menv_dict\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mcls\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mload_env_dict\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mfilename\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mload_from_package\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mload_from_package\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m\u001b[0;32m    108\u001b[0m \u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m    109\u001b[0m         \u001b[0mllGrid\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0menv_dict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;34m\"grid\"\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
+   "execution_count": null,
-      "\u001b[1;32mc:\\users\\u216993\\.conda\\envs\\flatland3-rl\\lib\\site-packages\\flatland\\envs\\persistence.py\u001b[0m in \u001b[0;36mload_env_dict\u001b[1;34m(cls, filename, load_from_package)\u001b[0m\n\u001b[0;32m    143\u001b[0m         \u001b[1;32melif\u001b[0m \u001b[0mfilename\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mendswith\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"pkl\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m    144\u001b[0m             \u001b[1;32mtry\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m--> 145\u001b[1;33m                 \u001b[0menv_dict\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mpickle\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mloads\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mload_data\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m\u001b[0;32m    146\u001b[0m             \u001b[1;32mexcept\u001b[0m \u001b[0mValueError\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m    147\u001b[0m                 \u001b[0mprint\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"pickle failed to load file:\"\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mfilename\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;34m\" trying msgpack (deprecated)...\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
+   "metadata": {},
-      "\u001b[1;31mAttributeError\u001b[0m: Can't get attribute 'RailAgentStatus' on <module 'flatland.envs.agent_utils' from 'c:\\\\users\\\\u216993\\\\.conda\\\\envs\\\\flatland3-rl\\\\lib\\\\site-packages\\\\flatland\\\\envs\\\\agent_utils.py'>"
+   "outputs": [],
-     ]
-    }
-   ],
   "source": [
-    "env, env_dict = RailEnvPersister.load_new(\"test_env_loop.pkl\", load_from_package=\"env_data.tests\")\n",
    "oRT = RenderTool(env)\n",
-    "oRT.render_env()\n",
+    "oRT.render_env()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "scrolled": false
+   },
+   "outputs": [],
+   "source": [
    "image_arr = oRT.get_image()\n",
    "canvas = ipycanvas.Canvas()\n",
    "canvas.put_image_data(image_arr)\n",
    "display(canvas)\n",
    "\n",
-    "for iEp in range(5):\n",
+    "done={\"__all__\":False}\n",
-    "    env.reset(False, False, True)\n",
+    "while not done[\"__all__\"]:\n",
-    "    env._max_episode_steps = 100\n",
+    "    actions = {}\n",
-    "\n",
+    "    for agent_handle, agents in enumerate(env.agents):\n",
-    "    done={\"__all__\":False}\n",
+    "        actions.update({agent_handle:rea.MOVE_FORWARD})\n",
-    "    while not done[\"__all__\"]:\n",
+    "    obs, rew, done, info = env.step(actions)\n",
-    "        obs, rew, done, info = env.step(action_dict_={int(0):rea.MOVE_FORWARD})\n",
+    "    oRT.render_env()\n",
-    "        oRT.render_env()\n",
+    "    gIm = oRT.get_image()\n",
-    "        gIm = oRT.get_image()\n",
+    "    canvas.put_image_data(gIm)\n",
-    "        canvas.put_image_data(gIm)\n",
+    "    time.sleep(0.1)"
-    "        time.sleep(0.1)"
   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
  }
 ],
 "metadata": {

 %% 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("test_env_loop.pkl", load_from_package="env_data.tests")
+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.render_env()
+```
+%% Cell type:code id: tags:
+``` python
 image_arr = oRT.get_image()
 canvas = ipycanvas.Canvas()
 canvas.put_image_data(image_arr)
 display(canvas)
-for iEp in range(5):
+done={"__all__":False}
-    env.reset(False, False, True)
+while not done["__all__"]:
-    env._max_episode_steps = 100
+    actions = {}
+    for agent_handle, agents in enumerate(env.agents):
-    done={"__all__":False}
+        actions.update({agent_handle:rea.MOVE_FORWARD})
-    while not done["__all__"]:
+    obs, rew, done, info = env.step(actions)
-        obs, rew, done, info = env.step(action_dict_={int(0):rea.MOVE_FORWARD})
+    oRT.render_env()
-        oRT.render_env()
+    gIm = oRT.get_image()
-        gIm = oRT.get_image()
+    canvas.put_image_data(gIm)
-        canvas.put_image_data(gIm)
+    time.sleep(0.1)
-        time.sleep(0.1)
+```
-```
+%% Cell type:code id: tags:
-%% Output
+``` python
-    ---------------------------------------------------------------------------
+```
-    AttributeError                            Traceback (most recent call last)
-    <ipython-input-3-24ea30f53e36> in <module>
-    ----> 1 env, env_dict = RailEnvPersister.load_new("test_env_loop.pkl", load_from_package="env_data.tests")
-          2 oRT = RenderTool(env)
-          3 oRT.render_env()
-          4 image_arr = oRT.get_image()
-          5 canvas = ipycanvas.Canvas()
-    c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\flatland\envs\persistence.py in load_new(cls, filename, load_from_package)
-        105     def load_new(cls, filename, load_from_package=None):
-        106
-    --> 107         env_dict = cls.load_env_dict(filename, load_from_package=load_from_package)
-        108
-        109         llGrid = env_dict["grid"]
-    c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\flatland\envs\persistence.py in load_env_dict(cls, filename, load_from_package)
-        143         elif filename.endswith("pkl"):
-        144             try:
-    --> 145                 env_dict = pickle.loads(load_data)
-        146             except ValueError:
-        147                 print("pickle failed to load file:", filename, " trying msgpack (deprecated)...")
-    AttributeError: Can't get attribute 'RailAgentStatus' on <module 'flatland.envs.agent_utils' from 'c:\\users\\u216993\\.conda\\envs\\flatland3-rl\\lib\\site-packages\\flatland\\envs\\agent_utils.py'>