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

Fixed passing arguments to new wrapper structure

Browse Source
This commit is contained in:
ottofabian 2021-06-24 17:19:26 +02:00
parent e250ec491e
commit 3c774382b7
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
alr_envs/utils/make_env_helpers.py
View File

@ -20,10 +20,14 @@ def make_contextual_env(env_id, context, seed, rank):
def make_dmp_env(**kwargs): def make_dmp_env(**kwargs):
name = kwargs.pop("name") name = kwargs.pop("name")
_env = gym.make(name) _env = gym.make(name)
return DmpWrapper(_env, **kwargs) for wrapper in kwargs.pop("wrappers"):
_env = wrapper(_env)
return DmpWrapper(_env, **kwargs.get("mp_kwargs"))
def make_detpmp_env(**kwargs): def make_detpmp_env(**kwargs):
name = kwargs.pop("name") name = kwargs.pop("name")
_env = gym.make(name) _env = gym.make(name)
return DetPMPWrapper(_env, **kwargs) for wrapper in kwargs.pop("wrappers"):
_env = wrapper(_env)
return DetPMPWrapper(_env, **kwargs.get("mp_kwargs"))