Hotfixing master tests. (#13282)
This commit is contained in:
parent
59c378d069
commit
662b143b71
|
@ -31,7 +31,7 @@ class TextClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTestC
|
|||
tf_model_mapping = TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING
|
||||
|
||||
@require_torch
|
||||
def test_pt_bert_small(self):
|
||||
def test_small_model_pt(self):
|
||||
text_classifier = pipeline(
|
||||
task="text-classification", model="Narsil/tiny-distilbert-sequence-classification", framework="pt"
|
||||
)
|
||||
|
@ -40,7 +40,7 @@ class TextClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTestC
|
|||
self.assertEqual(nested_simplify(outputs), [{"label": "LABEL_1", "score": 0.502}])
|
||||
|
||||
@require_tf
|
||||
def test_tf_bert_small(self):
|
||||
def test_small_model_tf(self):
|
||||
text_classifier = pipeline(
|
||||
task="text-classification", model="Narsil/tiny-distilbert-sequence-classification", framework="tf"
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue