Fix bart base test (#6587)

2020-08-18 21:28:10 -04:00
parent 1529bf9680
commit ab42d74850
2 changed files with 4 additions and 5 deletions
--- a/tests/test_modeling_marian.py
+++ b/tests/test_modeling_marian.py
@@ -205,9 +205,9 @@ class TestMarian_MT_EN(MarianIntegrationTest):
        self._assert_generated_batch_equal_expected()


-class TestMarian_eng_zho(MarianIntegrationTest):
-    src = "eng"
-    tgt = "zho"
+class TestMarian_en_zh(MarianIntegrationTest):
+    src = "en"
+    tgt = "zh"
    src_text = ["My name is Wolfgang and I live in Berlin"]
    expected_text = ["我叫沃尔夫冈 我住在柏林"]