fix layer tests

evkotov · Apr 5, 2024 · 7043559 · 7043559
1 parent d0b33d0
commit 7043559
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/layer_tests/common/layer_test_class.py b/tests/layer_tests/common/layer_test_class.py
@@ -168,8 +168,11 @@ def compare_ie_results_with_framework(self, infer_res, framework_res, framework_
         from common.utils.common_utils import allclose
         for framework_out_name in framework_res:
             ie_out_name = framework_out_name
-            if ie_out_name not in infer_res and len(infer_res) == 1:
-                ie_res = list(infer_res.values())[0]
+            if ie_out_name not in infer_res:
+                if len(infer_res) == 1:
+                    ie_res = list(infer_res.values())[0]
+                else:
+                    continue
             else:
                 ie_res = infer_res[ie_out_name]