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

Fix reacher link

Browse Source
This commit is contained in:
kngwyu 2021-04-19 18:42:06 +09:00
parent a2ba3511cd
commit 62188b5867
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mujoco_maze/reacher.py
View File

@ -1,7 +1,7 @@
""" """
Based on the reacher in `dm_control`_. Based on the reacher in `dm_control`_.
.. _gym: https://github.com/openai/gym .. _dm_control: https://github.com/deepmind/dm_control
""" """
from typing import Tuple from typing import Tuple
@ -54,10 +54,14 @@ class ReacherEnv(AgentModel):
def reset_model(self) -> np.ndarray: def reset_model(self) -> np.ndarray:
qpos = self.init_qpos + self.np_random.uniform( qpos = self.init_qpos + self.np_random.uniform(
low=-0.1, high=0.1, size=self.model.nq, low=-0.1,
high=0.1,
size=self.model.nq,
) )
qvel = self.init_qvel + self.np_random.uniform( qvel = self.init_qvel + self.np_random.uniform(
low=-0.1, high=0.1, size=self.model.nv, low=-0.1,
high=0.1,
size=self.model.nv,
) )
self.set_state(qpos, qvel) self.set_state(qpos, qvel)