From d47d98cb1bc49b4d87a0558631286666f67e9f7b Mon Sep 17 00:00:00 2001 From: Allen Goodman Date: Mon, 8 Apr 2024 16:03:35 -0400 Subject: [PATCH] rotation --- .../_rotations/test__quaternion_mean.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 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 index 4256855edd..01d3b1baac 100644 --- 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, )