fix bugs (#5046)

PaddleNLP/examples/text_generation/couplet/data.py

@@ -104,7 +104,7 @@ class CoupletDataset(TranslationDataset):
                 '`train`, `dev` or `test` is supported but `{}` is passed in'.
                 format(mode))
         # Download data
-        self.data = self.get_data(root=root)
+        self.data = self.get_data(mode=mode, root=root)
         self.vocab, _ = self.get_vocab(root)
         self.transform()
 

PaddleNLP/paddlenlp/datasets/translation.py

@@ -266,7 +266,7 @@ class IWSLT15(TranslationDataset):
                 raise ValueError("`transform_func` must have length of two for"
                                  "source and target.")
         # Download data and read data
-        self.data = self.get_data(root=root)
+        self.data = self.get_data(mode=mode, root=root)
 
         if transform_func is not None:
             self.data = [(transform_func[0](data[0]),