From 4c791c9eb44d68034e4e8c669a253c6786e9269b Mon Sep 17 00:00:00 2001
From: heya5 <yanhe.ian@outlook.com>
Date: Sun, 19 Feb 2023 09:44:27 +0800
Subject: [PATCH] fix datatype

---
 parlai/tasks/wizard_of_internet/agents.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/parlai/tasks/wizard_of_internet/agents.py b/parlai/tasks/wizard_of_internet/agents.py
index 4f1eda40547..8dfe952b55c 100644
--- a/parlai/tasks/wizard_of_internet/agents.py
+++ b/parlai/tasks/wizard_of_internet/agents.py
@@ -150,7 +150,7 @@ class WizardOfInternetBaseTeacher(DialogTeacher):
 
     def __init__(self, opt: Opt, shared=None):
         opt = deepcopy(opt)
-        self.datatype = get_dtype(opt)
+        self.datatype = opt.get('datatype', 'train')
         opt['datafile'] = _path(opt)
         self.include_persona = opt.get('include_persona', CONST.INCLUDE_PERSONA_DEFAULT)
         self.skip_empty_text = opt.get(