fixed a bug in prismatic joint (#568)

facebookresearch · Jul 6, 2023 · a81b419 · a81b419
1 parent 52851f2
commit a81b419
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchkin/torchkin/joint.py b/torchkin/torchkin/joint.py
@@ -363,7 +363,7 @@ def _check_input(angle: torch.Tensor):
 
     def translation(self, angle: torch.Tensor) -> torch.Tensor:
         _PrismaticJointImpl._check_input(angle)
-        return self.translation(angle.view(-1, 1))
+        return self._translation_impl(angle.view(-1, 1))
 
     def _relative_pose_impl(self, angle: torch.Tensor) -> torch.Tensor:
         trans = self.translation(angle)