diff --git a/tests/layer_tests/pytorch_tests/test_sum.py b/tests/layer_tests/pytorch_tests/test_sum.py index 040bc813fb0..e8f18f36826 100644 --- a/tests/layer_tests/pytorch_tests/test_sum.py +++ b/tests/layer_tests/pytorch_tests/test_sum.py @@ -9,7 +9,8 @@ from pytorch_layer_test_class import PytorchLayerTest class TestSum(PytorchLayerTest): def _prepare_input(self): import numpy as np - return (np.random.randn(1, 3, 224, 224).astype(np.float32),) + # This test had sporadically failed by accuracy. Try to resolve that by using int numbers in input + return (np.random.randint(-10, 10, (1, 3, 224, 224)).astype(np.float32),) def create_model(self, axes, keep_dims): @@ -37,4 +38,5 @@ class TestSum(PytorchLayerTest): @pytest.mark.nightly @pytest.mark.precommit def test_sum(self, axes, keep_dim, ie_device, precision, ir_version): - self._test(*self.create_model(axes, keep_dim), ie_device, precision, ir_version) + self._test(*self.create_model(axes, keep_dim), + ie_device, precision, ir_version)