Always download data on interactive wizard. (facebookresearch#3079)

analysmith · Sep 14, 2020 · 5f18bc2 · 5f18bc2
1 parent a3f581c
commit 5f18bc2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/parlai/tasks/wizard_of_wikipedia/worlds.py b/parlai/tasks/wizard_of_wikipedia/worlds.py
@@ -18,6 +18,8 @@
 from parlai.tasks.self_chat.worlds import SelfChatWorld as SelfChatBaseWorld
 from parlai.utils.misc import warn_once
 
+from .build import build
+
 from projects.wizard_of_wikipedia.knowledge_retriever.knowledge_retriever import (
     KnowledgeRetrieverAgent,
 )
@@ -74,6 +76,7 @@ def _set_up_knowledge_agent(self, add_token_knowledge=False):
 
     def _load_topics(self, opt):
         # Load possible chosen topics
+        build(opt)
         topics_path = os.path.join(
             opt['datapath'], 'wizard_of_wikipedia', 'topic_splits.json'
         )