Moving translation pipeline to new testing scheme. (#13297)

* Moving `translation` pipeline to new testing scheme. * Update tokenization mbart tests.
2021-08-27 12:26:17 +02:00
parent 319d840b46
commit a3f96f366a
4 changed files with 85 additions and 25 deletions
--- a/tests/test_tokenization_mbart.py
+++ b/tests/test_tokenization_mbart.py
@@ -235,7 +235,9 @@ class MBartEnroIntegrationTest(unittest.TestCase):

    @require_torch
    def test_tokenizer_translation(self):
-        inputs = self.tokenizer._build_translation_inputs("A test", src_lang="en_XX", tgt_lang="ar_AR")
+        inputs = self.tokenizer._build_translation_inputs(
+            "A test", return_tensors="pt", src_lang="en_XX", tgt_lang="ar_AR"
+        )

        self.assertEqual(
            nested_simplify(inputs),