diff --git a/tests/layer_tests/pytorch_tests/test_expand.py b/tests/layer_tests/pytorch_tests/test_expand.py index 4ef275ae0d4..ff9640ef706 100644 --- a/tests/layer_tests/pytorch_tests/test_expand.py +++ b/tests/layer_tests/pytorch_tests/test_expand.py @@ -117,7 +117,7 @@ class TestExpandAs(PytorchLayerTest): class TestDynamicExpand(PytorchLayerTest): def _prepare_input(self): import numpy as np - last_dym = random.randint(1,4) + last_dym = random.randint(2,8) return (np.random.randn(1, 3, 1).astype(np.float32), last_dym) def create_model(self, dim):