[TESTS] Fix rerun mechanism in case of failed tracing (#26758)

### Details: - *Fix rerun mechanism in case of failed tracing* ### Tickets: - *CVS-153111* Co-authored-by: Andrei Kochin <[email protected]>
openvinotoolkit · Sep 25, 2024 · 416bfb4 · 416bfb4
1 parent 094f1bc
commit 416bfb4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/layer_tests/py_frontend_tests/test_torchvision_preprocessor.py b/tests/layer_tests/py_frontend_tests/test_torchvision_preprocessor.py
@@ -36,7 +36,7 @@ def _infer_pipelines(test_input, preprocess_pipeline, input_channels=3):
         try:
             return _infer_pipelines_impl(test_input, preprocess_pipeline, input_channels)
         except RuntimeError as e:
-            if "builtin cannot be used as a value" in e:
+            if "builtin cannot be used as a value" in str(e):
                 # This is a potentially sporadic issue
                 print(f"An error occurred: {e}. Retrying...")
                 retries += 1

diff --git a/tests/layer_tests/pytorch_tests/pytorch_layer_test_class.py b/tests/layer_tests/pytorch_tests/pytorch_layer_test_class.py
@@ -76,7 +76,7 @@ def _test(self, model, ref_net, kind, ie_device, precision, ir_version, infer_ti
             try:
                 return self._test_impl(model, ref_net, kind, ie_device, precision, ir_version, infer_timeout, dynamic_shapes, **kwargs)
             except RuntimeError as e:
-                if "builtin cannot be used as a value" in e:
+                if "builtin cannot be used as a value" in str(e):
                     # This is a potentially sporadic issue
                     print(f"An error occurred: {e}. Retrying...")
                     retries += 1