From 2cf44246bb9df5e64f2b012368519609e78f7018 Mon Sep 17 00:00:00 2001 From: Georgy Krivoruchko Date: Fri, 5 Jul 2024 16:12:52 +0400 Subject: [PATCH] Added a missing change in ir_engine.py --- tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py | 6 ++---- 1 file changed, 2 insertions(+), 4 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 index 08af7e68f551b6..aec4049c6b32e8 100644 --- a/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py +++ b/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py @@ -35,8 +35,7 @@ def read_rt_info_attr(elem): val_dict = {} for child in elem: child_val = read_rt_info_attr(child) - node_name = child.attrib.get('name', child.tag) - val_dict[node_name] = child_val + val_dict[child.attrib.get('name', child.tag)] = child_val return val_dict @@ -105,8 +104,7 @@ def __load_xml(self): statistics[layer.find('name').text] = {'min': layer.find('min').text, 'max': layer.find('max').text} elif child.tag == 'rt_info': for elem in child: - node_name = elem.attrib['name'] if elem.attrib['name'] else elem.tag - self.meta_data[node_name] = read_rt_info_attr(elem) + self.meta_data[elem.attrib.get('name', elem.tag)] = read_rt_info_attr(elem) # TODO: Remove this part when POT updates to using of rt_info elif child.tag == 'quantization_parameters':