Skip to content

Commit 4d06979

Browse files
committed
fix failing test due to wrong inverse function call
1 parent 5e0f624 commit 4d06979

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/pytorch_kinematics/transforms/transform3d.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,7 @@ def transform_normals(self, normals):
405405
if normals.dim() not in [2, 3]:
406406
msg = "Expected normals to have dim = 2 or dim = 3: got shape %r"
407407
raise ValueError(msg % (normals.shape,))
408-
mat = self._get_matrix_inverse()[:, :3, :3]
408+
mat = self.inverse().get_matrix()[:, :3, :3]
409409
normals_out = _broadcast_bmm(normals, mat)
410410

411411
# This doesn't pass unit tests. TODO investigate further
@@ -435,7 +435,7 @@ def transform_shape_operator(self, shape_operators):
435435
if shape_operators.dim() not in [3, 4]:
436436
msg = "Expected shape_operators to have dim = 3 or dim = 4: got shape %r"
437437
raise ValueError(msg % (shape_operators.shape,))
438-
mat = self._get_matrix_inverse()[:, :3, :3]
438+
mat = self.inverse().get_matrix()[:, :3, :3]
439439
shape_operators_out = _broadcast_bmm(mat.permute(0, 2, 1), _broadcast_bmm(shape_operators, mat))
440440

441441
# When transform is (1, 4, 4) and shape_operator is (P, 3, 3) return

0 commit comments

Comments
 (0)