fix: #27

cophi-wue · Nov 23, 2018 · d97ee7e · d97ee7e
1 parent d285f9e
commit d97ee7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cophi/utils.py b/src/cophi/utils.py
@@ -22,7 +22,7 @@ def construct_ngrams(tokens, n=2, sep=" "):
     return (sep.join(ngram)
             for ngram in zip(*(itertools.islice(i, token, None)
                              for token, i in enumerate(itertools.tee(tokens,
-                                                                     2)))))
+                                                                     n)))))
 
 
 def find_tokens(document, token_pattern=r"\p{L}+\p{P}?\p{L}+", maximum=None):