diff --git a/llm_web_kit/main_html_parser/parser/layout_batch_parser.py b/llm_web_kit/main_html_parser/parser/layout_batch_parser.py index 616d72e9..5259cea3 100644 --- a/llm_web_kit/main_html_parser/parser/layout_batch_parser.py +++ b/llm_web_kit/main_html_parser/parser/layout_batch_parser.py @@ -151,7 +151,7 @@ def find_blocks_drop(self, element, depth, element_dict, parent_keyy, parent_lab length_tail = len(element.tail.strip()) idd = element.get('id') tag = element.tag - layer_nodes = element_dict[depth] + layer_nodes = element_dict.get(depth, {}) class_tag = element.get('class') ori_keyy = (tag, class_tag, idd) if idd and idd.strip():