Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
662b143b
Unverified
Commit
662b143b
authored
Aug 26, 2021
by
Nicolas Patry
Committed by
GitHub
Aug 26, 2021
Browse files
Hotfixing master tests. (#13282)
parent
59c378d0
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
tests/test_pipelines_text_classification.py
tests/test_pipelines_text_classification.py
+2
-2
No files found.
tests/test_pipelines_text_classification.py
View file @
662b143b
...
@@ -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"
)
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment