Skip to content

Commit 5c80e3d

Browse files
committed
Ensure correct seeding behaviour. Using the self.np_random rather than the np.random. This makes sure:
1. Deterministic env resetting. 2. Different seeds for envs running in parallel, e.g. Vector Env of gymnasium.
1 parent c822f28 commit 5c80e3d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

metaworld/envs/mujoco/sawyer_xyz/sawyer_xyz_env.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -522,8 +522,10 @@ def evaluate_state(self, obs, action):
522522
raise NotImplementedError
523523

524524
def reset(self, seed=None, options=None):
525+
if seed:
526+
self.seed(seed)
525527
self.curr_path_length = 0
526-
obs, info = super().reset()
528+
obs, info = super().reset(seed=seed)
527529
self._prev_obs = obs[:18].copy()
528530
obs[18:36] = self._prev_obs
529531
obs = np.float64(obs)
@@ -544,7 +546,7 @@ def _get_state_rand_vec(self):
544546
assert self._last_rand_vec is not None
545547
return self._last_rand_vec
546548
else:
547-
rand_vec = np.random.uniform(
549+
rand_vec = self.np_random.uniform(
548550
self._random_reset_space.low,
549551
self._random_reset_space.high,
550552
size=self._random_reset_space.low.size,

0 commit comments

Comments
 (0)