Skip to content

Commit f1f198f

Browse files
WIP: Add dual-arm description tags
1 parent 85c0717 commit f1f198f

File tree

1 file changed

+19
-11
lines changed

1 file changed

+19
-11
lines changed

robot_descriptions/_descriptions.py

Lines changed: 19 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -53,13 +53,21 @@ def has_urdf(self) -> bool:
5353
DESCRIPTIONS: Dict[str, Description] = {
5454
"a1_description": Description(Format.URDF, tags={"quadruped"}),
5555
"a1_mj_description": Description(Format.MJCF, tags={"quadruped"}),
56-
"ability_hand_description": Description(Format.URDF),
57-
"ability_hand_mj_description": Description(Format.MJCF),
56+
"ability_hand_description": Description(
57+
Format.URDF, tags={"end_effector"}
58+
),
59+
"ability_hand_mj_description": Description(
60+
Format.MJCF, tags={"end_effector"}
61+
),
5862
"aliengo_description": Description(Format.URDF, tags={"quadruped"}),
5963
"aliengo_mj_description": Description(Format.MJCF, tags={"quadruped"}),
60-
"allegro_hand_description": Description(Format.URDF),
61-
"allegro_hand_mj_description": Description(Format.MJCF),
62-
"aloha_mj_description": Description(Format.MJCF),
64+
"allegro_hand_description": Description(
65+
Format.URDF, tags={"end_effector"}
66+
),
67+
"allegro_hand_mj_description": Description(
68+
Format.MJCF, tags={"end_effector"}
69+
),
70+
"aloha_mj_description": Description(Format.MJCF, tags={"dual_arm"}),
6371
"anymal_b_description": Description(Format.URDF, tags={"quadruped"}),
6472
"anymal_b_mj_description": Description(Format.MJCF, tags={"quadruped"}),
6573
"anymal_c_description": Description(Format.URDF, tags={"quadruped"}),
@@ -71,7 +79,7 @@ def has_urdf(self) -> bool:
7179
"b1_description": Description(Format.URDF, tags={"quadruped"}),
7280
"b2_description": Description(Format.URDF, tags={"quadruped"}),
7381
"barrett_hand_description": Description(Format.URDF),
74-
"baxter_description": Description(Format.URDF),
82+
"baxter_description": Description(Format.URDF, tags={"dual_arm"}),
7583
"berkeley_humanoid_description": Description(
7684
Format.URDF, tags={"humanoid"}
7785
),
@@ -82,12 +90,12 @@ def has_urdf(self) -> bool:
8290
"cf2_mj_description": Description(Format.MJCF),
8391
"double_pendulum_description": Description(Format.URDF),
8492
"draco3_description": Description(Format.URDF, tags={"humanoid"}),
85-
"edo_description": Description(Format.URDF),
93+
"edo_description": Description(Format.URDF, tags={"arm"}),
8694
"elf2_description": Description(Format.URDF, tags={"humanoid"}),
8795
"elf2_mj_description": Description(Format.MJCF, tags={"humanoid"}),
8896
"ergocub_description": Description(Format.URDF, tags={"humanoid"}),
8997
"eve_r3_description": Description(Format.URDF),
90-
"fanuc_m710ic_description": Description(Format.URDF),
98+
"fanuc_m710ic_description": Description(Format.URDF, tags={"arm"}),
9199
"fetch_description": Description(Format.URDF),
92100
"finger_edu_description": Description(Format.URDF),
93101
"g1_description": Description(Format.URDF, tags={"humanoid"}),
@@ -115,15 +123,15 @@ def has_urdf(self) -> bool:
115123
"leap_hand_mj_description": Description(Format.MJCF),
116124
"mini_cheetah_description": Description(Format.URDF, tags={"quadruped"}),
117125
"minitaur_description": Description(Format.URDF, tags={"quadruped"}),
118-
"nextage_description": Description(Format.URDF),
126+
"nextage_description": Description(Format.URDF, tags={"dual_arm"}),
119127
"op3_mj_description": Description(Format.MJCF, tags={"humanoid"}),
120128
"panda_description": Description(Format.URDF),
121129
"panda_mj_description": Description(Format.MJCF),
122130
"pepper_description": Description(Format.URDF),
123131
"piper_description": Description(Format.URDF),
124132
"piper_mj_description": Description(Format.MJCF),
125133
"poppy_ergo_jr_description": Description(Format.URDF),
126-
"poppy_torso_description": Description(Format.URDF),
134+
"poppy_torso_description": Description(Format.URDF, tags={"dual_arm"}),
127135
"pr2_description": Description(Format.URDF),
128136
"r2_description": Description(Format.URDF, tags={"humanoid"}),
129137
"reachy_description": Description(Format.URDF),
@@ -160,7 +168,7 @@ def has_urdf(self) -> bool:
160168
"viper_mj_description": Description(Format.MJCF),
161169
"widow_mj_description": Description(Format.MJCF),
162170
"xarm7_mj_description": Description(Format.MJCF),
163-
"yumi_description": Description(Format.URDF),
171+
"yumi_description": Description(Format.URDF, tags={"dual_arm"}),
164172
"z1_description": Description(Format.URDF),
165173
"z1_mj_description": Description(Format.MJCF),
166174
}

0 commit comments

Comments
 (0)