"docs/source/vscode:/vscode.git/clone" did not exist on "153d1361c7dcc91c7735cae73e1f594cfcab3e21"
Unverified Commit 662b143b authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Hotfixing master tests. (#13282)

parent 59c378d0
...@@ -31,7 +31,7 @@ class TextClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTestC ...@@ -31,7 +31,7 @@ class TextClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTestC
tf_model_mapping = TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING tf_model_mapping = TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING
@require_torch @require_torch
def test_pt_bert_small(self): def test_small_model_pt(self):
text_classifier = pipeline( text_classifier = pipeline(
task="text-classification", model="Narsil/tiny-distilbert-sequence-classification", framework="pt" task="text-classification", model="Narsil/tiny-distilbert-sequence-classification", framework="pt"
) )
...@@ -40,7 +40,7 @@ class TextClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTestC ...@@ -40,7 +40,7 @@ class TextClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTestC
self.assertEqual(nested_simplify(outputs), [{"label": "LABEL_1", "score": 0.502}]) self.assertEqual(nested_simplify(outputs), [{"label": "LABEL_1", "score": 0.502}])
@require_tf @require_tf
def test_tf_bert_small(self): def test_small_model_tf(self):
text_classifier = pipeline( text_classifier = pipeline(
task="text-classification", model="Narsil/tiny-distilbert-sequence-classification", framework="tf" task="text-classification", model="Narsil/tiny-distilbert-sequence-classification", framework="tf"
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment