riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix non-keyworded varargs handling in DefaultArgumentHandler for pipeline.

Browse Source
This commit is contained in:
Morgan Funtowicz 2019-12-16 15:49:09 +01:00
parent 9c391277cc
commit bbc707cf39
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
transformers/pipelines.py
View File

@ -55,7 +55,12 @@ class DefaultArgumentHandler(ArgumentHandler):
return kwargs['X']
elif 'data' in kwargs:
return kwargs['data']
elif len(args) > 0:
elif len(args) == 1:
if isinstance(args[0], list):
return args[0]
else:
return [args[0]]
elif len(args) > 1:
return list(args)
raise ValueError('Unable to infer the format of the provided data (X=, data=, ...)')
@ -240,7 +245,7 @@ class NerPipeline(Pipeline):
def __call__(self, *texts, **kwargs):
inputs, answers = self._args_parser(*texts, **kwargs), []
for sentence in texts:
for sentence in inputs:
# Ugly token to word idx mapping (for now)
token_to_word, words = [], sentence.split(' ')