Commit 4efa6676 authored by metataro's avatar metataro

Merge remote-tracking branch 'origin/sparse-and-deadlock-rewards' into shortest-path-obs

# Conflicts:
#	envs/flatland/utils/gym_env_wrappers.py
#	envs/flatland_sparse.py
parents 974e97bd 453cc631
......@@ -430,6 +430,7 @@ class FlatlandRenderWrapper(RailEnv,gym.Env):
if not self.renderer:
self.initialize_renderer(mode=mode)
return self.update_renderer(mode=mode)
def initialize_renderer(self, mode="human"):
......@@ -463,3 +464,4 @@ class FlatlandRenderWrapper(RailEnv,gym.Env):
# Finally the env is closed when RLLib exits but at that time there is no window
# and hence the error
print("Could Not close window due to:",e)
......@@ -16,6 +16,7 @@ from envs.flatland.utils.gym_env_wrappers import AvailableActionsWrapper, SkipNo
DeadlockWrapper, ShortestPathActionWrapper, DeadlockResolutionWrapper
from envs.flatland_base import FlatlandBase
class FlatlandSparse(FlatlandBase):
def __init__(self, env_config) -> None:
......
Markdown is supported
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