Internal change

h-joo · copybara-github · commit 6c84e8a75dbb · 2025-04-29T23:56:11.000-07:00
PiperOrigin-RevId: 753039066
Change-Id: I8c054de816a922426da58993036dc9e626472ad0
diff --git a/mujoco_playground/_src/locomotion/h1/joystick.py b/mujoco_playground/_src/locomotion/h1/joystick.py
@@ -149,7 +149,7 @@ def step(self, state: mjx_env.State, action: jax.Array) -> mjx_env.State:
     motor_targets = self._default_pose + action * self._config.action_scale
     motor_targets = jp.clip(motor_targets, self._lowers, self._uppers)
     data = mjx_env.step(
-        self.mjx_model, state.data, motor_targets, self._n_frames
+        self.mjx_model, state.data, motor_targets, self._n_frames  # pytype: disable=attribute-error
     )
 
     obs = self._get_obs(data, state.info, state.obs, noise_rng)

Original file line number	Diff line number	Diff line change
`@@ -149,7 +149,7 @@ def step(self, state: mjx_env.State, action: jax.Array) -> mjx_env.State:`
`149`	`149`	`motor_targets = self._default_pose + action * self._config.action_scale`
`150`	`150`	`motor_targets = jp.clip(motor_targets, self._lowers, self._uppers)`
`151`	`151`	`data = mjx_env.step(`
`152`		`- self.mjx_model, state.data, motor_targets, self._n_frames`
	`152`	`+ self.mjx_model, state.data, motor_targets, self._n_frames # pytype: disable=attribute-error`
`153`	`153`	`)`
`154`	`154`
`155`	`155`	`obs = self._get_obs(data, state.info, state.obs, noise_rng)`