omission

d8b43600 · VictorSanh · Victor SANH · ee5a6856 · d8b43600
Commit d8b43600 authored Feb 07, 2020 by VictorSanh Committed by Victor SANH Feb 07, 2020
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/test_pipelines.py tests/test_pipelines.py +2 -2

No files found.
--- a/tests/test_pipelines.py
+++ b/tests/test_pipelines.py
@@ -10,13 +10,13 @@ from .utils import require_tf, require_torch
 QA_FINETUNED_MODELS = {
    ("bert-base-uncased", "bert-large-uncased-whole-word-masking-finetuned-squad", None),
    ("bert-base-cased", "bert-large-cased-whole-word-masking-finetuned-squad", None),
-    ("bert-base-uncased", "distilbert-base-cased-distilled-squad", None),
+    ("bert-base-cased", "distilbert-base-cased-distilled-squad", None),
 }

 TF_QA_FINETUNED_MODELS = {
    ("bert-base-uncased", "bert-large-uncased-whole-word-masking-finetuned-squad", None),
    ("bert-base-cased", "bert-large-cased-whole-word-masking-finetuned-squad", None),
-    ("bert-base-uncased", "distilbert-base-cased-distilled-squad", None),
+    ("bert-base-cased", "distilbert-base-cased-distilled-squad", None),
 }

 TF_NER_FINETUNED_MODELS = {