Only parse sentences in constructor if chain argument not given

Otherwise, cannot construct from json when saved with retain_original=False
jsvine · Sep 11, 2017 · dcafe08 · dcafe08
1 parent 3d86fbd
commit dcafe08
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/markovify/text.py b/markovify/text.py
@@ -35,7 +35,8 @@ def __init__(self, input_text, state_size=2, chain=None, parsed_sentences=None,
             self.rejoined_text = self.sentence_join(map(self.word_join, self.parsed_sentences))
             self.chain = chain or Chain(self.parsed_sentences, state_size)
         else:
-            parsed = parsed_sentences or self.generate_corpus(input_text)
+            if not chain:
+                parsed = parsed_sentences or self.generate_corpus(input_text)
             self.chain = chain or Chain(parsed, state_size)
 
     def to_dict(self):