Notebook is working (Flatland3)

0de93b9a · adrian_egli2 · 68c99627 · 0de93b9a
Commit 0de93b9a authored 2 years ago by adrian_egli2
--- a/notebooks/test-saved-envs.ipynb
+++ b/notebooks/test-saved-envs.ipynb
@@ -69,7 +69,7 @@
   "outputs": [],
   "source": [
    "#lsDirs = [ \"../env_data/railway\", \"../env_data/tests\"]\n",
-    "lsPackages = [ \"env_data.railway\", \"env_data.tests\"]"
+    "lsPackages = [ \"env_data.railway\"]"
   ]
  },
  {
@@ -146,7 +146,7 @@
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
-   "version": "3.6.7"
+   "version": "3.6.8"
  },
  "latex_envs": {
   "LaTeX_envs_menu_present": true,

 %% Cell type:markdown id: tags:

 ## Load some (old) env files to check they work
 This notebook just loads some old env files, renders them, and runs a few steps.
 This is just a sanity check that these old envs will still load.
 Many of them use deprecated data formats so it's just so that we can avoid deleting them for now, and so new participants are not confused by us shipping env files which don't work...

 %% Cell type:code id: tags:

 ``` python
 %load_ext autoreload
 %autoreload 2
 ```

 %% Cell type:code id: tags:

 ``` python
 from IPython import display
 display.display(display.HTML("<style>.container { width:95% !important; }</style>"))
 ```

 %% Cell type:code id: tags:

 ``` python
 import PIL
 import glob
 import pickle
 import msgpack
 import importlib_resources as ir
 ```

 %% Cell type:code id: tags:

 ``` python
 from flatland.envs.persistence import RailEnvPersister
 ```

 %% Cell type:code id: tags:

 ``` python
 from flatland.utils import env_edit_utils as eeu
 from flatland.utils import jupyter_utils as ju
 ```

 %% Cell type:code id: tags:

 ``` python
 #lsDirs = [ "../env_data/railway", "../env_data/tests"]
-lsPackages = [ "env_data.railway", "env_data.tests"]
+lsPackages = [ "env_data.railway"]
 ```

 %% Cell type:code id: tags:

 ``` python
 ltPackRes = []
 for sPack in lsPackages:
    for sExt in ["mpk", "pkl"]:
        #lsFiles += glob.glob(sDir + "/*" + sExt)
        ltPackRes += [ (sPack, sRes) for sRes in ir.contents(sPack) if sRes.endswith(sExt) ]
 ltPackRes
 ```

 %% Cell type:code id: tags:

 ``` python
 if False:
    for sFile in lsFiles:
        try:
            with open(sFile, "rb") as fIn:
                env_dict = pickle.load(fIn)
            print("pickle:", sFile)
        except ValueError as oErr:
            try:
                with open(sFile, "rb") as fIn:
                    env_dict = msgpack.load(fIn)
                print("msgpack: ", sFile)
            except ValueError as oErr:
                print("msgpack failed: ", sFile)


 ```

 %% Cell type:code id: tags:

 ``` python
 for sPack, sRes in ltPackRes:
    print("Loading: ", sPack, sRes)
    env, env_dict = RailEnvPersister.load_new(sRes, load_from_package=sPack)
    env.reset()
    oCanvas = ju.EnvCanvas(env, ju.AlwaysForward(env))
    oCanvas.show()
    for iStep in range(10):
        oCanvas.step()
        oCanvas.render()
 ```