Unit test for dynamic expand fixed (#24684)

### Tickets:
 - CVS-142090
This commit is contained in:
Mustafa Cavus 2024-05-25 09:37:08 -07:00 committed by GitHub
parent 33c786cb2c
commit 0fe524bcbe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -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):