Commit 7d22fefd authored by Julien Chaumond's avatar Julien Chaumond
Browse files

[pipeline] Alias NerPipeline as TokenClassificationPipeline

parent 61a2b7dc
...@@ -101,6 +101,7 @@ from .pipelines import ( ...@@ -101,6 +101,7 @@ from .pipelines import (
PipelineDataFormat, PipelineDataFormat,
QuestionAnsweringPipeline, QuestionAnsweringPipeline,
TextClassificationPipeline, TextClassificationPipeline,
TokenClassificationPipeline,
pipeline, pipeline,
) )
from .tokenization_albert import AlbertTokenizer from .tokenization_albert import AlbertTokenizer
......
...@@ -649,6 +649,9 @@ class NerPipeline(Pipeline): ...@@ -649,6 +649,9 @@ class NerPipeline(Pipeline):
return answers return answers
TokenClassificationPipeline = NerPipeline
class QuestionAnsweringArgumentHandler(ArgumentHandler): class QuestionAnsweringArgumentHandler(ArgumentHandler):
""" """
QuestionAnsweringPipeline requires the user to provide multiple arguments (i.e. question & context) to be mapped QuestionAnsweringPipeline requires the user to provide multiple arguments (i.e. question & context) to be mapped
......
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