From d61bb6429d15f68db8d2d30001e419c75dc77da6 Mon Sep 17 00:00:00 2001 From: MooYeh Date: Mon, 18 Mar 2024 10:19:37 +0800 Subject: [PATCH] catch json loads exception --- .../ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py index 1e82f62fde..0175905676 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py @@ -25,6 +25,7 @@ import sys import time import zlib import json +from json.decoder import JSONDecodeError from datetime import datetime, timezone from functools import wraps from pathlib import Path @@ -398,7 +399,11 @@ def is_summary_compare(input_param): def is_md5_compare(input_parma): with FileOpen(input_parma.get("npu_pkl_path"), "r") as npu_pkl: - line = json.loads(npu_pkl.readline()) + pkl_lines = npu_pkl.readline() + try: + line = json.loads(pkl_lines) + except JSONDecodeError as err: + raise CompareException(CompareException.INVALID_FILE_ERROR) from err if len(line) < 3: return False if line[2]: -- Gitee