Skip to content
Snippets Groups Projects
Commit beb27f18 authored by adrian_egli2's avatar adrian_egli2
Browse files

flatland3 ready

parent 928e92b3
No related branches found
No related tags found
No related merge requests found
%% Cell type:markdown id: tags:
# Plot the Env and some trees
%% Cell type:code id: tags:
``` python
%load_ext autoreload
%autoreload 2
```
%% Cell type:code id: tags:
``` python
import jpy_canvas
import random
import time
import sys
```
%% Cell type:code id: tags:
``` python
# in case you need to tweak your PYTHONPATH...
sys.path.append("../flatland")
```
%% Cell type:code id: tags:
``` python
import flatland.core.env
import flatland.utils.rendertools as rt
from flatland.envs.rail_env import RailEnv
from flatland.envs.rail_generators import sparse_rail_generator
from flatland.envs.line_generators import sparse_line_generator
from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.predictions import ShortestPathPredictorForRailEnv
```
%% Output
---------------------------------------------------------------------------
KeyboardInterrupt Traceback (most recent call last)
<ipython-input-4-94a42146e075> in <module>
1 import flatland.core.env
2 import flatland.utils.rendertools as rt
----> 3 from flatland.envs.rail_env import RailEnv
4 from flatland.envs.rail_generators import sparse_rail_generator
5 from flatland.envs.line_generators import sparse_line_generator
c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\flatland\envs\rail_env.py in <module>
23 from flatland.envs.timetable_generators import timetable_generator
24 from flatland.envs import persistence
---> 25 from flatland.envs import agent_chains as ac
26 from flatland.envs.fast_methods import fast_position_equal
27
c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\flatland\envs\agent_chains.py in <module>
1
----> 2 import networkx as nx
3 import numpy as np
4
5 from typing import List, Tuple
c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\networkx\__init__.py in <module>
59 from networkx.relabel import *
60
---> 61 import networkx.generators
62 from networkx.generators import *
63
c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\networkx\generators\__init__.py in <module>
12 from networkx.generators.ego import *
13 from networkx.generators.expanders import *
---> 14 from networkx.generators.geometric import *
15 from networkx.generators.internet_as_graphs import *
16 from networkx.generators.intersection import *
c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\networkx\generators\geometric.py in <module>
8
9 try:
---> 10 from scipy.spatial import cKDTree as KDTree
11 except ImportError:
12 _is_scipy_available = False
c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\scipy\spatial\__init__.py in <module>
97 from .ckdtree import *
98 from .qhull import *
---> 99 from ._spherical_voronoi import SphericalVoronoi
100 from ._plotutils import *
101 from ._procrustes import procrustes
c:\users\u216993\.conda\envs\flatland3-rl\lib\site-packages\scipy\spatial\_spherical_voronoi.py in <module>
15 import numpy as np
16 import scipy
---> 17 from . import _voronoi
18 from scipy.spatial import cKDTree
19
c:\users\u216993\.conda\envs\flatland3-rl\lib\importlib\_bootstrap.py in _find_and_load(name, import_)
c:\users\u216993\.conda\envs\flatland3-rl\lib\importlib\_bootstrap.py in _find_and_load_unlocked(name, import_)
c:\users\u216993\.conda\envs\flatland3-rl\lib\importlib\_bootstrap.py in _find_spec(name, path, target)
c:\users\u216993\.conda\envs\flatland3-rl\lib\importlib\_bootstrap_external.py in find_spec(cls, fullname, path, target)
c:\users\u216993\.conda\envs\flatland3-rl\lib\importlib\_bootstrap_external.py in _get_spec(cls, fullname, path, target)
c:\users\u216993\.conda\envs\flatland3-rl\lib\importlib\_bootstrap_external.py in find_spec(self, fullname, target)
c:\users\u216993\.conda\envs\flatland3-rl\lib\importlib\_bootstrap_external.py in _path_stat(path)
KeyboardInterrupt:
%% Cell type:code id: tags:
``` python
from IPython.core.display import display, HTML
display(HTML("<style>.container { width:90% !important; }</style>"))
```
%% Cell type:markdown id: tags:
# Generate
%% Cell type:code id: tags:
``` python
nAgents = 3
n_cities = 2
max_rails_between_cities = 2
max_rails_in_city = 4
seed = 0
env = RailEnv(
width=20,
height=30,
rail_generator=sparse_rail_generator(
max_num_cities=n_cities,
seed=seed,
grid_mode=True,
max_rails_between_cities=max_rails_between_cities,
max_rail_pairs_in_city=max_rails_in_city
),
line_generator=sparse_line_generator(),
number_of_agents=nAgents,
obs_builder_object=TreeObsForRailEnv(max_depth=3, predictor=ShortestPathPredictorForRailEnv())
)
init_observation = env.reset()
```
%% Cell type:markdown id: tags:
# Render
%% Cell type:code id: tags:
``` python
oRT = rt.RenderTool(env,gl="PILSVG")
oRT.render_env(show_observations=False,show_predictions=True)
img = oRT.get_image()
```
%% Cell type:code id: tags:
``` python
jpy_canvas.Canvas(img)
```
%% Cell type:code id: tags:
``` python
oRT = rt.RenderTool(env,gl="PIL")
oRT.render_env(show_observations=False,show_predictions=True)
img = oRT.get_image()
```
%% Cell type:code id: tags:
``` python
jpy_canvas.Canvas(img)
```
......
source diff could not be displayed: it is too large. Options to address this: view the blob.
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