[fix] prosocial dialogue format error

This commit is contained in:
theblackcat102
2023-01-22 14:00:20 +00:00
parent 98bf14884d
commit 736f46fb00
@@ -67,4 +67,4 @@ class ProsocialDialogue(Dataset):
return len(self.pairs)
def __getitem__(self, idx):
return format_pair(self.pairs[idx])
return self.pairs[idx]