Fixed #7

Co-Evolve · Mar 25, 2024 · 3d15ca7 · 3d15ca7
1 parent aa60ca3
commit 3d15ca7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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}",