From a2948322f9d37f53d9a63edaf1932a926be26537 Mon Sep 17 00:00:00 2001
From: bobloy <alboblexloy@gmail.com>
Date: Fri, 6 Nov 2020 12:42:03 -0500
Subject: [PATCH] Download ubuntu data to the cog data directory

---
 chatter/chat.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/chatter/chat.py b/chatter/chat.py
index d63a664..4e3400c 100644
--- 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