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

Fix: Some simple_reacher did not seed correctly

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-08-28 18:32:11 +02:00
parent 07aeb779a7
commit 820e781a0c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
fancy_gym/envs/classic_control/simple_reacher/simple_reacher.py
View File

@ -45,8 +45,9 @@ class SimpleReacherEnv(BaseReacherTorqueEnv):
def reset(self, *, seed: Optional[int] = None, options: Optional[Dict[str, Any]] = None) \ def reset(self, *, seed: Optional[int] = None, options: Optional[Dict[str, Any]] = None) \
-> Tuple[ObsType, Dict[str, Any]]: -> Tuple[ObsType, Dict[str, Any]]:
ret = super().reset(seed=seed, options=options)
self._generate_goal() self._generate_goal()
return super().reset(seed=seed, options=options) return ret
def _get_reward(self, action: np.ndarray): def _get_reward(self, action: np.ndarray):
diff = self.end_effector - self._goal diff = self.end_effector - self._goal