Skip to content
Snippets Groups Projects
Commit feb9357b authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

Update render_agent_behavior.py

parent 3725068b
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
torch_training/render_agent_behavior.py
+ 1
− 1
View file @ feb9357b
...@@ -3,7 +3,7 @@ from collections import deque ...@@ -3,7 +3,7 @@ from collections import deque
import numpy as np import numpy as np
import torch import torch
from flatland.envs.malfunction_generators import malfunction_from_params from flatland.envs.malfunction_generators import malfunction_from_params, MalfunctionParameters
from flatland.envs.observations import TreeObsForRailEnv from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.rail_env import RailEnv from flatland.envs.rail_env import RailEnv
from flatland.envs.rail_generators import sparse_rail_generator from flatland.envs.rail_generators import sparse_rail_generator
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment