From db915e756ae45880492252757cdab123e2cda389 Mon Sep 17 00:00:00 2001 From: westfish Date: Mon, 17 Oct 2022 16:32:18 +0800 Subject: [PATCH] Bug fix uie utils (#3490) * fix bug the result variable is not defined * update --- model_zoo/uie/utils.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/model_zoo/uie/utils.py b/model_zoo/uie/utils.py index 1e1f2930f86b..0b6b5475ddae 100644 --- a/model_zoo/uie/utils.py +++ b/model_zoo/uie/utils.py @@ -222,17 +222,16 @@ def reader(data_path, max_seq_len=512): if len(content) <= max_content_len: yield json_line else: - if result['end'] - result['start'] > max_content_len: - logger.warn( - "result['end '] - result ['start'] exceeds max_content_len, which will result in no valid instance being returned" - ) result_list = json_line['result_list'] json_lines = [] accumulate = 0 while True: cur_result_list = [] - for result in result_list: + if result['end'] - result['start'] > max_content_len: + logger.warning( + "result['end'] - result ['start'] exceeds max_content_len, which will result in no valid instance being returned" + ) if result['start'] + 1 <= max_content_len < result[ 'end'] and result['end'] - result[ 'start'] <= max_content_len: