fixed merge conflict

mryzhov · Oct 26, 2023 · 2d38d21 · 2d38d21
1 parent fbf0ed1
commit 2d38d21
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/layer_tests/pytorch_tests/test_adaptive_avg_pool.py b/tests/layer_tests/pytorch_tests/test_adaptive_avg_pool.py
@@ -33,8 +33,8 @@ def forward(self, input_tensor):
     @pytest.mark.precommit
     @pytest.mark.precommit_ts_backend
     @pytest.mark.precommit_fx_backend
-    def test_adaptive_avg_pool3d(self, ie_device, precision, ir_version, input_shape, output_size):
-        self.input_tensor = np.random.randn(*input_shape).astype(np.float32)
+    def test_adaptive_avg_pool3d(self, ie_device, precision, ir_version, input_tensor, output_size):
+        self.input_tensor = np.random.randn(*input_tensor).astype(np.float32)
         self._test(*self.create_model(output_size), ie_device, precision, ir_version)
 
 

diff --git a/tests/layer_tests/pytorch_tests/test_repeat_interleave.py b/tests/layer_tests/pytorch_tests/test_repeat_interleave.py
@@ -39,7 +39,7 @@ def forward(self, input_tensor):
     def test_repeat_interleave_const_repeats(self, ie_device, precision, ir_version, input_data):
         repeats = input_data['repeats']
         if type(repeats) is list and len(repeats) == 1:
-            repeats = random.randint(1, 5) for _ in range(repeats[0])
+            repeats = [random.randint(1, 5) for _ in range(repeats[0])]
 
         dim = input_data['dim']
         self._test(*self.create_model_const_repeat(repeats, dim),