Apply suggestions from code review

Co-authored-by: Roman Kazantsev <roman.kazantsev@intel.com>
This commit is contained in:
Maxim Vafin 2023-10-04 22:55:11 +02:00 committed by GitHub
parent e65ee445b5
commit 68fef6243c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -73,13 +73,13 @@ class TestConvertModel:
"OpenVINO outputs does not contain tensor with name {}".format(out_name)
cur_ov_res = ov_outputs[out_name]
print(f"fw_re: {cur_fw_res};\n ov_res: {cur_ov_res}")
is_ok = is_ok or compare_two_tensors(cur_ov_res, cur_fw_res, fw_eps)
is_ok = is_ok and compare_two_tensors(cur_ov_res, cur_fw_res, fw_eps)
else:
for i in range(len(ov_outputs)):
cur_fw_res = fw_outputs[i]
cur_ov_res = ov_outputs[i]
print(f"fw_res: {cur_fw_res};\n ov_res: {cur_ov_res}")
is_ok = is_ok or compare_two_tensors(cur_ov_res, cur_fw_res, fw_eps)
is_ok = is_ok and compare_two_tensors(cur_ov_res, cur_fw_res, fw_eps)
assert is_ok, "Accuracy validation failed"
def teardown_method(self):