@@ -1055,7 +1055,12 @@ def reset_joints_by_scale(
1055
1055
joint_vel = joint_vel .clamp_ (- joint_vel_limits , joint_vel_limits )
1056
1056
1057
1057
# set into the physics simulation
1058
- asset .write_joint_state_to_sim (joint_pos , joint_vel , env_ids = env_ids , joint_ids = asset_cfg .joint_ids )
1058
+ asset .write_joint_state_to_sim (
1059
+ joint_pos .view (len (env_ids ), - 1 ),
1060
+ joint_vel .view (len (env_ids ), - 1 ),
1061
+ env_ids = env_ids ,
1062
+ joint_ids = asset_cfg .joint_ids
1063
+ )
1059
1064
1060
1065
1061
1066
def reset_joints_by_offset (
@@ -1089,7 +1094,12 @@ def reset_joints_by_offset(
1089
1094
joint_vel = joint_vel .clamp_ (- joint_vel_limits , joint_vel_limits )
1090
1095
1091
1096
# set into the physics simulation
1092
- asset .write_joint_state_to_sim (joint_pos , joint_vel , env_ids = env_ids , joint_ids = asset_cfg .joint_ids )
1097
+ asset .write_joint_state_to_sim (
1098
+ joint_pos .view (len (env_ids ), - 1 ),
1099
+ joint_vel .view (len (env_ids ), - 1 ),
1100
+ env_ids = env_ids ,
1101
+ joint_ids = asset_cfg .joint_ids
1102
+ )
1093
1103
1094
1104
1095
1105
def reset_nodal_state_uniform (
0 commit comments