Apply suggestions from code review

Co-authored-by: Maxim Vafin <[email protected]>
mryzhov · Oct 26, 2023 · fbf0ed1 · fbf0ed1
1 parent 466dc9b
commit fbf0ed1
Showing 1 changed file 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
@@ -8,7 +8,7 @@
 from pytorch_layer_test_class import PytorchLayerTest
 
 
-@pytest.mark.parametrize('input_shape', [[1, 2, 8, 9, 10], [2, 8, 9, 10]])
+@pytest.mark.parametrize('input_tensor', [[1, 2, 8, 9, 10], [2, 8, 9, 10]])
 @pytest.mark.parametrize('output_size', [[5, 7, 9], 7])
 class TestAdaptiveAvgPool3D(PytorchLayerTest):
 
@@ -22,8 +22,8 @@ def __init__(self, output_size) -> None:
                 super().__init__()
                 self.output_size = output_size
 
-            def forward(self, input_shape):
-                return torch.nn.functional.adaptive_avg_pool3d(input_shape, self.output_size)
+            def forward(self, input_tensor):
+                return torch.nn.functional.adaptive_avg_pool3d(input_tensor, self.output_size)
 
         ref_net = None