From f15238a37eeb09c495eff194f3963aed28909b82 Mon Sep 17 00:00:00 2001 From: "Brian S. Stephan" Date: Sat, 28 Jul 2012 14:01:05 -0500 Subject: [PATCH] Markov: abort new chain tack-on if even that's giving us __stop --- modules/Markov.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/modules/Markov.py b/modules/Markov.py index 7721e66..cb05395 100644 --- a/modules/Markov.py +++ b/modules/Markov.py @@ -400,9 +400,12 @@ class Markov(Module): key_hits = self._retrieve_chains_for_key(self.start1, self.start2, context_id) gen_words.append(self._get_suitable_word_from_choices(key_hits, gen_words, min_size)) - # new word 2 - key_hits = self._retrieve_chains_for_key(self.start2, found_word, context_id) - gen_words.append(self._get_suitable_word_from_choices(key_hits, gen_words, min_size)) + if gen_words[-1] == self.stop: + break + else: + # new word 2 + key_hits = self._retrieve_chains_for_key(self.start2, gen_words[-1], context_id) + gen_words.append(self._get_suitable_word_from_choices(key_hits, gen_words, min_size)) # chop off the seed data at the start gen_words = gen_words[2:]