Skip to content

Commit 4b7a9e4

Browse files
btabacopybara-github
authored andcommitted
Fix #187.
PiperOrigin-RevId: 800351136 Change-Id: I70507b55bb086d1c5754faac0aca14b9045253f5
1 parent 63c863d commit 4b7a9e4

File tree

1 file changed

+2
-2
lines changed
  • mujoco_playground/_src/manipulation/franka_emika_panda_robotiq

1 file changed

+2
-2
lines changed

mujoco_playground/_src/manipulation/franka_emika_panda_robotiq/push_cube.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -511,8 +511,8 @@ def _get_single_obs(self, data: mjx.Data, info: dict[str, Any]) -> jax.Array:
511511
obj_quat_w_noise = math.quat_mul(rand_quat, obj_quat)
512512
obj_pos = data.xpos[self._obj_body]
513513
obj_pos_w_noise = obj_pos + jax.random.uniform(
514-
key3, minval=0, maxval=self._config.noise_config.noise_scales.obj_pos
515-
)
514+
key3, (3,), minval=-1, maxval=1
515+
) * self._config.noise_config.noise_scales.obj_pos
516516

517517
# Add noise to robot proprio observation.
518518
info["rng"], key1, key2, key3, key4 = jax.random.split(info["rng"], 5)

0 commit comments

Comments
 (0)