Fix accidental change #2

sshleifer · May 8, 2020 · 158cd06 · 158cd06
1 parent c38e824
commit 158cd06
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py
@@ -43,9 +43,21 @@
     ("distilbert-base-cased", "distilbert-base-cased", None),
 }
 
-TF_TEXT_CLASSIF_FINETUNED_MODELS = {("bert-base-uncased", "distilbert-base-uncased-finetuned-sst-2-english",)}
+TF_TEXT_CLASSIF_FINETUNED_MODELS = {
+    (
+        "bert-base-uncased",
+        "distilbert-base-uncased-finetuned-sst-2-english",
+        "distilbert-base-uncased-finetuned-sst-2-english",
+    )
+}
 
-TEXT_CLASSIF_FINETUNED_MODELS = {("distilbert-base-cased", "distilbert-base-uncased-finetuned-sst-2-english",)}
+TEXT_CLASSIF_FINETUNED_MODELS = {
+    (
+        "distilbert-base-cased",
+        "distilbert-base-uncased-finetuned-sst-2-english",
+        "distilbert-base-uncased-finetuned-sst-2-english",
+    )
+}
 
 TEXT_GENERATION_FINETUNED_MODELS = {
     ("gpt2", "gpt2"),
@@ -320,7 +332,7 @@ def test_tf_fill_mask(self):
 
     @require_torch
     def test_torch_summarization(self):
-        tlong = "=" * 102570
+        tlong = "=" * 2000
         valid_inputs = ["A string like this", ["list of strings entry 1", "list of strings v2"], tlong]
         invalid_inputs = [4, "<mask>"]
         mandatory_keys = ["summary_text"]