Fix joint friction/armature set events #2439
Conversation
asuzukii
requested review from
kellyguo11,
Mayankm96 and
jsmith-bdai
as code owners
| @@ -574,7 +574,7 @@ def randomize_joint_parameters( | |||
| distribution=distribution, | |||
| ) | |||
| asset.write_joint_friction_coefficient_to_sim( | |||
| friction_coeff[env_ids[:, None], joint_ids], joint_ids=joint_ids, env_ids=env_ids | |||
| friction_coeff[env_ids, joint_ids], joint_ids=joint_ids, env_ids=env_ids | |||
There was a problem hiding this comment.
This fix won't work if you have a subset of env ids and joint ids over which the term is operating. There is definitely a bug here if joint_ids is a slice.
The way to fix it is to check if one of them is a slice. If yes, we keep the broadcasting. Otherwise we index directly.
| func=mdp.randomize_joint_parameters, | ||
| mode="startup", | ||
| params={ | ||
| "asset_cfg": SceneEntityCfg("robot", body_names=".*"), |
There was a problem hiding this comment.
Try to make joint_names as a subset of joints to see if the above fix still works.
There was a problem hiding this comment.
I see, this seems like an issue because I'm specifying body_names and not joint_names.
I can close this if that is correct
There was a problem hiding this comment.
randomize_joint_parameters should work with joint_names instead of body_names, since it modifies the joints in the articulations instead of bodies.
Description
Please include a summary of the change and which issue is fixed. Please also include relevant motivation and context.
List any dependencies that are required for this change.
Fixes #2440, which was a bug that passed in a wrong dim tensor into the joint friction/armature attribute
Adds a dummy joint rand event in anymalc env cfg to make sure this doesn't break again.
Type of change
Screenshots
Please attach before and after screenshots of the change if applicable.
Checklist
pre-commitchecks with./isaaclab.sh --formatconfig/extension.tomlfileCONTRIBUTORS.mdor my name already exists there