Skip to content

Commit a79193a

Browse files
Handle MJCF descriptions in Pinocchio loader
See #62 (comment)
1 parent 851eee6 commit a79193a

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

Diff for: robot_descriptions/loaders/pinocchio.py

+11-5
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,15 @@ def load_robot_description(
5353
module = import_module(f"robot_descriptions.{description_name}")
5454
if commit is not None:
5555
os.environ.pop("ROBOT_DESCRIPTION_COMMIT", None)
56-
robot = pin.RobotWrapper.BuildFromURDF(
57-
filename=module.URDF_PATH,
58-
package_dirs=get_package_dirs(module),
59-
root_joint=root_joint,
60-
)
56+
if hasattr(module, "URDF_PATH"):
57+
robot = pin.RobotWrapper.BuildFromURDF(
58+
filename=module.URDF_PATH,
59+
package_dirs=get_package_dirs(module),
60+
root_joint=root_joint,
61+
)
62+
elif hasattr(module, "MJCF_PATH"):
63+
robot = pin.RobotWrapper.BuildFromMJCF(
64+
filename=module.MJCF_PATH,
65+
root_joint=root_joint,
66+
)
6167
return robot

0 commit comments

Comments
 (0)