adding better chunking for txt2kg

pyg-team · Dec 10, 2024 · e15d908 · e15d908
1 parent 6c2a09f
commit e15d908
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/torch_geometric/nn/nlp/txt2kg.py b/torch_geometric/nn/nlp/txt2kg.py
@@ -182,8 +182,9 @@ def get_num_procs():
 
 
 def chunk_text(text: str, chunk_size: int = 512) -> list[str]:
-    """Function to chunk text into sentence-based segments.
-    Co-authored with Claude AI
+    """
+    Function to chunk text into sentence-based segments.
+    Co-authored with Claude AI.
     """
     # If the input text is empty or None, return an empty list
     if not text: