Less iterations in doctest and fix #34

DARIAH-DE · Dec 21, 2017 · e9ff28b · e9ff28b
1 parent 1bd4652
commit e9ff28b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dariah_topics/mallet.py b/dariah_topics/mallet.py
@@ -383,7 +383,9 @@ def train_topics(self, mallet_binary, cleanup=False, **kwargs):
             >>> document_labels = ['document_label']
             >>> Mallet = Mallet(corpus_output='.')
             >>> mallet_corpus = Mallet.import_tokenized_corpus(tokenized_corpus, document_labels)
-            >>> mallet_topics = Mallet.train_topics(mallet_corpus, output_model='model.mallet')
+            >>> mallet_topics = Mallet.train_topics(mallet_corpus,
+            ...                                     output_model='model.mallet',
+            ...                                     num_iterations=10)
             >>> os.path.exists('model.mallet')
             True
         """