Skip to content

Commit f6a47c7

Browse files
committed
Add ngram parameter to make function work correctly
1 parent 7b97edc commit f6a47c7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def clean_text(text, ngrams_number: int = None):
2020
filtered_words_list = set
2121

2222
for word in text:
23-
cleaned_word = set(ngrams(_text_cleaner(word), 2))
23+
cleaned_word = set(ngrams(_text_cleaner(word), int(ngrams_number)))
2424
filtered_words_list = filtered_words_list.union(cleaned_word)
2525

2626
filtered_unique_words_list = [" ".join(i) for i in filtered_words_list]

0 commit comments

Comments
 (0)