rotation

Genentech · Apr 8, 2024 · d47d98c · d47d98c
1 parent c1a03fe
commit d47d98c
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/tests/beignet/ops/_geometry/_transformations/_rotations/test__quaternion_mean.py b/tests/beignet/ops/_geometry/_transformations/_rotations/test__quaternion_mean.py
@@ -17,12 +17,18 @@ def _strategy(function):
 
     return (
         {
-            "input": torch.from_numpy(rotation.as_quat(canonical=False)),
+            "input": torch.from_numpy(
+                rotation.as_quat(
+                    canonical=False,
+                ),
+            ),
         },
         torch.unsqueeze(
             torch.abs(
                 torch.from_numpy(
-                    rotation.mean().as_quat(canonical=False),
+                    rotation.mean().as_quat(
+                        canonical=False,
+                    ),
                 ),
             ),
             dim=0,
@@ -35,6 +41,10 @@ def test_quaternion_mean(data):
     parameters, expected = data
 
     torch.testing.assert_close(
-        torch.abs(beignet.ops.quaternion_mean(**parameters)),
+        torch.abs(
+            beignet.ops.quaternion_mean(
+                **parameters,
+            ),
+        ),
         expected,
     )