Enable self_chat to seed messages from tasks in parlai_internal (#3852)

* Enable self_chat to seed messages from tasks in parlai_internal * Update parlai/tasks/self_chat/worlds.py Co-authored-by: Stephen Roller <roller@fb.com>
facebookresearch · Jul 23, 2021 · 59774ab · 59774ab
1 parent 5b71567
commit 59774ab
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/parlai/tasks/self_chat/worlds.py b/parlai/tasks/self_chat/worlds.py
@@ -15,7 +15,11 @@
 
 
 def load_openers(opt) -> Optional[List[str]]:
-    base_task = opt['task'].split(':')[0]
+    if opt['task'].startswith('internal:') or opt['task'].startswith('fb:'):
+        base_task = opt['task']
+    else:
+        base_task = opt['task'].split(':')[0]
+
     if base_task == 'self_chat':
         # TODO(#2284): Load default openers from s3
         return None