def set_corpus(self, corpus): if isinstance(corpus, str): self.corpus = corpora.MmCorpus(corpus) elif isinstance(corpus, corpora.MmCorpus): self.corpus = corpus