fix unit test

openvinotoolkit · Aug 28, 2024 · 4e1b810 · 4e1b810
1 parent 3163800
commit 4e1b810
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/core/data/transform_libs/test_torchvision.py b/tests/unit/core/data/transform_libs/test_torchvision.py
@@ -915,6 +915,6 @@ def test_forward(self, keypoint_det_entity) -> None:
         )
         results = transform(deepcopy(keypoint_det_entity))
 
-        assert torch.all(results.bbox_info.center == torch.Tensor([[3.5, 3.5]]))
-        assert torch.all(results.bbox_info.scale == torch.Tensor([[8.75, 8.75]]))
+        assert np.array_equal(results.bbox_info.center, np.array([3.5, 3.5]))
+        assert np.array_equal(results.bbox_info.scale, np.array([8.75, 8.75]))
         assert results.keypoints.shape == (4, 2)