From 00ef09162e616722990e2724718777b0478de10d Mon Sep 17 00:00:00 2001 From: "Beomsu Kim (Shuki)" <123bskim@naver.com> Date: Wed, 7 Oct 2020 14:59:32 +0900 Subject: [PATCH] Fix convai2:normalized NormalizedTeacher --- parlai/tasks/convai2/agents.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/parlai/tasks/convai2/agents.py b/parlai/tasks/convai2/agents.py index 7bf1daa66ac..468bacdc120 100644 --- a/parlai/tasks/convai2/agents.py +++ b/parlai/tasks/convai2/agents.py @@ -108,7 +108,7 @@ def setup_data(self, path): for (text, labels, reward, candidates), new_episode in super().setup_data(path): text = self.normalize_replies(text) labels = [self.normalize_replies(l) for l in labels] - candidates = [self.normalize_replies(l) for l in labels] + candidates = [self.normalize_replies(c) for c in candidates] yield (text, labels, reward, candidates), new_episode