Fixed #2554 (#2619)

piskvorky · Oct 4, 2019 · 25f8a42 · 25f8a42
1 parent 26f1e81
commit 25f8a42
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gensim/corpora/wikicorpus.py b/gensim/corpora/wikicorpus.py
@@ -700,7 +700,7 @@ def get_texts(self):
  logger.warn(
  "user terminated iteration over Wikipedia corpus after %i documents with %i positions "
  "(total %i articles, %i positions before pruning articles shorter than %i words)",
- articles, positions, articles_all, positions_all, ARTICLE_MIN_WORDS
+ articles, positions, articles_all, positions_all, self.article_min_tokens
  )
  except PicklingError as exc:
  raise_from(PicklingError('Can not send filtering function {} to multiprocessing, '
@@ -709,7 +709,7 @@ def get_texts(self):
  logger.info(
  "finished iterating over Wikipedia corpus of %i documents with %i positions "
  "(total %i articles, %i positions before pruning articles shorter than %i words)",
- articles, positions, articles_all, positions_all, ARTICLE_MIN_WORDS
+ articles, positions, articles_all, positions_all, self.article_min_tokens
  )
  self.length = articles # cache corpus length
  finally: