Fix unit tests (#2844)

openvinotoolkit · Jan 29, 2024 · 35d202e · 35d202e
1 parent f54eb1d
commit 35d202e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/unit/algo/visual_prompting/backbones/test_tiny_vit.py b/tests/unit/algo/visual_prompting/backbones/test_tiny_vit.py
@@ -20,7 +20,7 @@
 
 
 class TestConv2d_BN:  # noqa: N801
-    def setup(self) -> None:
+    def setup_method(self) -> None:
         self.conv2d_bn = Conv2d_BN(a=1, b=1)
 
     def test_init(self) -> None:
@@ -107,7 +107,7 @@ def test_forward(self) -> None:
 
 
 class TestTinyViTBlock:
-    def setup(self) -> None:
+    def setup_method(self) -> None:
         self.dim = 4
         self.input_resolution = (6, 6)
         self.num_heads = 1
@@ -131,7 +131,7 @@ def test_forward(self) -> None:
 
 
 class TestBasicLayer:
-    def setup(self) -> None:
+    def setup_method(self) -> None:
         self.dim = 4
         self.input_resolution = (6, 6)
         self.depth = 1
@@ -160,7 +160,7 @@ def test_extra_repr(self) -> None:
 
 
 class TestTinyViT:
-    def setup(self) -> None:
+    def setup_method(self) -> None:
         self.tiny_vit = TinyViT(
             img_size=1024,
             embed_dims=[64, 128, 160, 320],