From cabe848f23807e69f3fbb3d8fda1cf64467cb138 Mon Sep 17 00:00:00 2001 From: Mikhail Ryzhov Date: Tue, 24 Oct 2023 21:43:01 +0200 Subject: [PATCH] fixed merge conflict --- tests/layer_tests/pytorch_tests/test_adaptive_avg_pool.py | 4 ++-- tests/layer_tests/pytorch_tests/test_repeat_interleave.py | 2 +- 2 files changed, 3 insertions(+), 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 index c572f1d8594672..bedd9beb0d4bcf 100644 --- 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 index a442491cf5e1c9..3df1ae1b421c02 100644 --- 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),