From 662b143b71eb5ef775e27a8f79798bb28b3283bd Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Thu, 26 Aug 2021 16:09:53 +0200 Subject: [PATCH] Hotfixing master tests. (#13282) --- tests/test_pipelines_text_classification.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_pipelines_text_classification.py b/tests/test_pipelines_text_classification.py index 90d6bae4b6..3daad11597 100644 --- a/tests/test_pipelines_text_classification.py +++ b/tests/test_pipelines_text_classification.py @@ -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" )