From 73eecabe061efba4f2a02ff25ad4715e71a46581 Mon Sep 17 00:00:00 2001 From: Qu0rise Date: Tue, 15 Apr 2025 17:24:35 +0900 Subject: [PATCH] Fix(argo2): Restore compatibility with modern kornia versions (>=0.7.0) --- pcdet/datasets/argo2/argo2_utils/so3.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/pcdet/datasets/argo2/argo2_utils/so3.py b/pcdet/datasets/argo2/argo2_utils/so3.py index f5ec5e095..edfe1cb53 100644 --- a/pcdet/datasets/argo2/argo2_utils/so3.py +++ b/pcdet/datasets/argo2/argo2_utils/so3.py @@ -16,9 +16,7 @@ def quat_to_mat(quat_wxyz: Tensor) -> Tensor: Returns: (...,3,3) 3D rotation matrices. """ - return C.quaternion_to_rotation_matrix( - quat_wxyz, order=C.QuaternionCoeffOrder.WXYZ - ) + return C.quaternion_to_rotation_matrix(quat_wxyz) # @torch.jit.script @@ -31,9 +29,7 @@ def mat_to_quat(mat: Tensor) -> Tensor: Returns: (...,4) Scalar first quaternions. """ - return C.rotation_matrix_to_quaternion( - mat, order=C.QuaternionCoeffOrder.WXYZ - ) + return C.rotation_matrix_to_quaternion(mat) @torch.jit.script