Download ubuntu data to the cog data directory

4 years ago · a2948322f9
parent 419863b07a
commit a2948322f9
1 changed files with 3 additions and 1 deletions
--- a/chatter/chat.py
+++ b/chatter/chat.py
@ -159,7 +159,9 @@ class Chatter(Cog):
        return out

    def _train_ubuntu(self):
-        trainer = UbuntuCorpusTrainer(self.chatbot)
+        trainer = UbuntuCorpusTrainer(
+            self.chatbot, ubuntu_corpus_data_directory=cog_data_path(self) / "ubuntu_data"
+        )
        trainer.train()
        return True