From 3ae05d1b5fe0a14c8d9378b06c33335f8c3573ff Mon Sep 17 00:00:00 2001 From: Charles Lin Date: Tue, 30 Apr 2024 20:33:27 -0700 Subject: [PATCH] Add user attribute to and attachment frames --- dm_control/mjcf/schema.py | 2 +- dm_control/mjcf/schema.xml | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/dm_control/mjcf/schema.py b/dm_control/mjcf/schema.py index 8bd8cc53..50dff12c 100644 --- a/dm_control/mjcf/schema.py +++ b/dm_control/mjcf/schema.py @@ -215,7 +215,7 @@ def _attachment_frame_spec(is_world_attachment): body_spec = MUJOCO.children['worldbody'].children['body'] # 'name' and 'childclass' attributes are excluded. for attrib_name in ( - 'mocap', 'pos', 'quat', 'axisangle', 'xyaxes', 'zaxis', 'euler'): + 'mocap', 'pos', 'quat', 'axisangle', 'xyaxes', 'zaxis', 'euler', 'user'): frame_spec.attributes[attrib_name] = copy.deepcopy( body_spec.attributes[attrib_name]) diff --git a/dm_control/mjcf/schema.xml b/dm_control/mjcf/schema.xml index 2f2d7020..eeaf9543 100644 --- a/dm_control/mjcf/schema.xml +++ b/dm_control/mjcf/schema.xml @@ -1307,6 +1307,7 @@ +