Fix: Some simple_reacher did not seed correctly

2023-08-28 18:32:11 +02:00 · 2023-08-28 18:32:11 +02:00 · 820e781a0c
commit 820e781a0c
parent 07aeb779a7
1 changed files with 2 additions and 1 deletions
--- a/fancy_gym/envs/classic_control/simple_reacher/simple_reacher.py
+++ b/fancy_gym/envs/classic_control/simple_reacher/simple_reacher.py
@ -45,8 +45,9 @@ class SimpleReacherEnv(BaseReacherTorqueEnv):

    def reset(self, *, seed: Optional[int] = None, options: Optional[Dict[str, Any]] = None) \
            -> Tuple[ObsType, Dict[str, Any]]:
+        ret = super().reset(seed=seed, options=options)
        self._generate_goal()
-        return super().reset(seed=seed, options=options)
+        return ret

    def _get_reward(self, action: np.ndarray):
        diff = self.end_effector - self._goal