From 3ec5ef5576189039f80e5cbc46a255efdf22828b Mon Sep 17 00:00:00 2001 From: Retr0327 Date: Sat, 18 Mar 2023 17:37:18 +0800 Subject: [PATCH] Refactor `create_title_task` to handle empty segmented data --- ckip2tei/tei/tags/body/body.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/ckip2tei/tei/tags/body/body.py b/ckip2tei/tei/tags/body/body.py index 3366d82..eae0645 100644 --- a/ckip2tei/tei/tags/body/body.py +++ b/ckip2tei/tei/tags/body/body.py @@ -19,9 +19,13 @@ async def create_title_task(text_tag: ET.Element, data: str, author: str): """The create_title_task function handles the title tag creation.""" + segmented_data = await segment(data) + + if not segmented_data: + return None + title_tag = ET.SubElement(text_tag, "title", author=author) sentence_tag = ET.SubElement(title_tag, "s") - segmented_data = await segment(data) tagger = TitleTagger(sentence_tag, TitleCleaner(segmented_data)) await tagger.create()