Refactor create_title_task to handle empty segmented data

Taiwan-Social-Media-Corpus · Mar 18, 2023 · 3ec5ef5 · 3ec5ef5
1 parent 91af58c
commit 3ec5ef5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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()