diff --git a/source/isaaclab_newton/isaaclab_newton/actuators/actuator_pd.py b/source/isaaclab_newton/isaaclab_newton/actuators/actuator_pd.py index 3694d8ddab58..1b8bcde0d3ce 100644 --- a/source/isaaclab_newton/isaaclab_newton/actuators/actuator_pd.py +++ b/source/isaaclab_newton/isaaclab_newton/actuators/actuator_pd.py @@ -154,7 +154,7 @@ def compute(self): # update the joint effort wp.launch( update_array2D_with_array2D_masked, - dim=(self._num_envs, self.num_joints), + dim=(self._num_envs, self._num_joints), inputs=[ self.data._actuator_effort_target, self.data.joint_effort, @@ -228,7 +228,7 @@ def compute(self): # update the joint effort wp.launch( update_array2D_with_array2D_masked, - dim=(self._num_envs, self.num_joints), + dim=(self._num_envs, self._num_joints), inputs=[ self.data._applied_effort, self.data.joint_effort,