rotation

Genentech · Apr 8, 2024 · c1a03fe · c1a03fe
1 parent 1e62ac2
commit c1a03fe
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/beignet/ops/_geometry/_transformations/_rotations/test__quaternion_mean.py b/tests/beignet/ops/_geometry/_transformations/_rotations/test__quaternion_mean.py
@@ -19,7 +19,14 @@ def _strategy(function):
         {
             "input": torch.from_numpy(rotation.as_quat(canonical=False)),
         },
-        torch.abs(torch.from_numpy(rotation.mean().as_quat(canonical=False))),
+        torch.unsqueeze(
+            torch.abs(
+                torch.from_numpy(
+                    rotation.mean().as_quat(canonical=False),
+                ),
+            ),
+            dim=0,
+        ),
     )