Fix typo

JasperHG90 · Jul 23, 2020 · 764a4b7 · 764a4b7
1 parent faa8828
commit 764a4b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/HAN/utils/preprocess.py b/HAN/utils/preprocess.py
@@ -25,7 +25,7 @@ def segmenter(input_texts: List[str]) -> List[List[str]]:
                 # Add to doc
                 doc.append(" ".join(current_sentence))
             txt_segmented.append(doc)
-    return doc
+    return txt_segmented
 
 # Input words output np arrays of vocabulary positions
 class Vectorizer(object):