Adding pipeline task aliases. (#11247)

* Adding task aliases and adding `token-classification` and
`text-classification` tasks.

* Cleaning docstring.
This commit is contained in:
Nicolas Patry
2021-04-15 09:51:24 +02:00
committed by GitHub
parent aaaed56ffc
commit c3fcba3219
5 changed files with 29 additions and 12 deletions

View File

@@ -17,7 +17,7 @@ import unittest
from .test_pipelines_common import MonoInputPipelineCommonMixin
class SentimentAnalysisPipelineTests(MonoInputPipelineCommonMixin, unittest.TestCase):
class TextClassificationPipelineTests(MonoInputPipelineCommonMixin, unittest.TestCase):
pipeline_task = "sentiment-analysis"
small_models = [
"sshleifer/tiny-distilbert-base-uncased-finetuned-sst-2-english"

View File

@@ -27,7 +27,7 @@ if is_torch_available():
VALID_INPUTS = ["A simple string", ["list of strings", "A simple string that is quite a bit longer"]]
class NerPipelineTests(CustomInputPipelineCommonMixin, unittest.TestCase):
class TokenClassificationPipelineTests(CustomInputPipelineCommonMixin, unittest.TestCase):
pipeline_task = "ner"
small_models = [
"sshleifer/tiny-dbmdz-bert-large-cased-finetuned-conll03-english"