dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add render_mode option to mp_wrapper

Browse Source
This commit is contained in:
Maximilian Huettenrauch 2021-04-30 16:10:16 +02:00
parent e035d9b621
commit 2c8335f632
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
alr_envs/utils/wrapper/mp_wrapper.py
View File

@ -17,6 +17,7 @@ class MPWrapper(gym.Wrapper, ABC):
post_traj_time: float = 0.,
policy_type: str = None,
weights_scale: float = 1.,
render_mode: str = None,
**mp_kwargs
):
super().__init__(env)
@ -36,7 +37,7 @@ class MPWrapper(gym.Wrapper, ABC):
self.policy = policy_class(env)
# rendering
self.render_mode = None
self.render_mode = render_mode
self.render_kwargs = None
# TODO: not yet final