Add kwargs to filter_dictionary

Richard Truong-Chau · Richard Truong-Chau · commit 5d4fc1fa8330 · 2020-11-09T14:12:53.000-06:00
diff --git a/data_describe/text/text_preprocessing.py b/data_describe/text/text_preprocessing.py
@@ -383,6 +383,6 @@ def filter_dictionary(
         corpus: Bag of Words (BoW) representation of documents (token_id, token_count).
     """
     dictionary = _compat["gensim"].corpora.Dictionary(text_docs)  # type: ignore
-    dictionary.filter_extremes(no_below=no_below, no_above=no_above)
+    dictionary.filter_extremes(no_below=no_below, no_above=no_above, **kwargs)
     corpus = [dictionary.doc2bow(doc) for doc in text_docs]
     return dictionary, corpus
