IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE TESTS] Add exception handling in merge_xmls script (#6442)

Browse Source
This commit is contained in:
Irina Efode 2021-06-29 17:03:06 +03:00 committed by GitHub
parent 4161bb7fa2
commit 1ca53717fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
inference-engine/tests/ie_test_utils/functional_test_utils/layer_tests_summary/merge_xmls.py
View File

@ -74,7 +74,15 @@ def merge_xml(input_folder_paths: list, output_folder_paths: str, output_filenam
xml_reports = glob.glob(os.path.join(folder_path, 'report*.xml'))
xml_root = ET.parse(xml_reports[0]).getroot()
xml_root = None
for xml_report in xml_reports:
try:
xml_root = ET.parse(xml_report).getroot()
break
except ET.ParseError:
logger.error(f'{xml_report} is incorrect! Error to get a xml root')
if xml_root is None:
logger.error(f'{folder_path} does not contain the correct xml files')
for op in xml_root.find("ops_list"):
if ops_list.find(op.tag) is None:
ET.SubElement(ops_list, op.tag)
@ -90,6 +98,8 @@ def merge_xml(input_folder_paths: list, output_folder_paths: str, output_filenam
with open(out_file_path, "w") as xml_file:
xml_file.write(ET.tostring(summary).decode('utf8'))
logger.info(f" Final report is saved to file: '{out_file_path}'")
if xml_root is None:
raise Exception("Error to make a XML root. Exit the app")
if __name__ == "__main__":