MO Patch reading custom rt info

openvinotoolkit · Sep 30, 2024 · 030da61 · 030da61
1 parent bcb3b79
commit 030da61
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py b/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py
@@ -539,10 +539,14 @@ def __read_old_api_map_element_type(attr, layer_type):
 
     @staticmethod
     def __read_rt_info_common(attr):
-        attr_name = attr.attrib['name']
-        version = int(attr.attrib['version'])
-        rt_info = OrderedDict()
-        for key in attr.attrib:
-            if key not in ('name', 'version'):
-                rt_info[key] = attr.attrib[key]
-        return {(attr_name, version): rt_info}
+        # ignore custom info, MO is deprecated so there's no needed for forward compatibility
+        if 'version' in attr.attrib:
+            attr_name = attr.attrib['name']
+            version = int(attr.attrib['version'])
+            rt_info = OrderedDict()
+            for key in attr.attrib:
+                if key not in ('name', 'version'):
+                    rt_info[key] = attr.attrib[key]
+            return {(attr_name, version): rt_info}
+        else:
+            return {}