Modify unit tests

openvinotoolkit · Jun 26, 2023 · 1b4ef0a · 1b4ef0a
1 parent 36ccfb1
commit 1b4ef0a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/unit/algorithms/detection/adapters/mmdet/models/heads/test_custom_dino_head.py b/tests/unit/algorithms/detection/adapters/mmdet/models/heads/test_custom_dino_head.py
@@ -171,10 +171,10 @@ def test_forward_train(self):
     @e2e_pytest_unit
     def test_simple_test_bboxes(self):
         feats = [
-            torch.randn([2, 256, 100, 134]),
-            torch.randn([2, 256, 50, 67]),
-            torch.randn([2, 256, 25, 34]),
-            torch.randn([2, 256, 13, 17]),
+            torch.zeros([2, 256, 100, 134]),
+            torch.zeros([2, 256, 50, 67]),
+            torch.zeros([2, 256, 25, 34]),
+            torch.zeros([2, 256, 13, 17]),
         ]
         img_metas = [
             {
@@ -208,6 +208,7 @@ def test_simple_test_bboxes(self):
         ]
         self.bbox_head.eval()
         results = self.bbox_head.simple_test_bboxes(feats, img_metas)
+        breakpoint()
         assert len(results) == 2
         assert results[0][0].shape == torch.Size([300, 5])
         assert results[0][1].shape == torch.Size([300])