IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PT FE]: fix sporaidic failure in pytorch layer tests (#15713)

Browse Source
This commit is contained in:
Ekaterina Aidova 2023-02-14 21:35:35 +04:00 committed by GitHub
parent 5cc1c2c5e5
commit 5ead83f315
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/layer_tests/pytorch_tests/pytorch_layer_test_class.py
View File

@ -128,7 +128,7 @@ class PytorchLayerTest:
for fw_tensor, ov_tensor in zip(flatten_fw_res, output_list):
if not isinstance(fw_tensor, torch.Tensor):
if np.isscalar(fw_tensor):
assert fw_tensor == np.array(ov_tensor).item()
assert fw_tensor == np.array(ov_tensor).item(), f"{fw_tensor} != {np.array(ov_tensor).item()}"
else:
if isinstance(fw_tensor, list):
ov_tensor = ov_tensor.tolist()

2
tests/layer_tests/pytorch_tests/test_bool.py
View File

@ -9,7 +9,7 @@ from pytorch_layer_test_class import PytorchLayerTest
class TestBool(PytorchLayerTest):
def _prepare_input(self):
import numpy as np
return (np.random.randn(1).astype(np.int32),)
return (np.random.randint(0, 10, 1).astype(np.int32),)
def create_model(self, input_type):
import torch