From 3d15ca7af4344406c1f2fa227141500b30d54158 Mon Sep 17 00:00:00 2001 From: dmarzoug Date: Mon, 25 Mar 2024 10:01:29 +0100 Subject: [PATCH] Fixed #7 --- moojoco/mjcf/component.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/moojoco/mjcf/component.py b/moojoco/mjcf/component.py index 37bcc20..e016322 100644 --- a/moojoco/mjcf/component.py +++ b/moojoco/mjcf/component.py @@ -7,6 +7,7 @@ import numpy as np from dm_control import mjcf from dm_control.mjcf import export_with_assets +from dm_control.mjcf.element import _AttachmentFrame from scipy.spatial.transform import Rotation @@ -35,7 +36,7 @@ def attach( position: Optional[np.ndarray] = None, euler: Optional[np.ndarray] = None, free_joint: bool = False, - ) -> None: + ) -> _AttachmentFrame: attachment_site = self.mjcf_body.add( "site", name=f"{self.base_name}_attachment_{other.base_name}",