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

Mitigation: Metaworld .reset ignores seeds; we must manually seed on reset.

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-08-28 17:57:24 +02:00
parent 22a72fed2f
commit 5921e0008b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fancy_gym/meta/metaworld_adapter.py
View File

@ -56,10 +56,10 @@ class MetaworldResetFix(gym.Wrapper, gym.utils.RecordConstructorArgs):
gym.Wrapper.__init__(self, env) gym.Wrapper.__init__(self, env)
def reset(self, **kwargs): def reset(self, **kwargs):
ret = self.env.reset(**kwargs) self.env.reset(**kwargs)
if 'seed' in kwargs: if 'seed' in kwargs:
self.env.seed(kwargs['seed']) self.env.seed(kwargs['seed'])
return ret return self.env.reset(**kwargs)
def make_metaworld(underlying_id: str, seed: int = 1, render_mode: Optional[str] = None, **kwargs): def make_metaworld(underlying_id: str, seed: int = 1, render_mode: Optional[str] = None, **kwargs):