Skip to content

Commit 88a4f85

Browse files
authored
[PyTorch Layer Tests] test_topK - Accuracy validation failed (#33194)
- Using a deterministic seed to generate test stimuli Ticket: CVS-153110
1 parent 920c54a commit 88a4f85

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/layer_tests/pytorch_tests/test_topk.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,5 +64,6 @@ def forward(self, input_tensor):
6464
@pytest.mark.precommit_torch_export
6565
@pytest.mark.precommit_fx_backend
6666
def test_topK(self, input_shape, k, dim, largest, sort, ie_device, precision, ir_version):
67-
self.input_tensor = np.random.randn(*input_shape).astype(np.float32)
67+
rng = np.random.default_rng(43)
68+
self.input_tensor = rng.normal(size=(input_shape)).astype(np.float32)
6869
self._test(*self.create_model(k, dim, largest, sort), ie_device, precision, ir_version)

0 commit comments

Comments
 (0)