Commit 43a4e1bb authored by Morgan Funtowicz's avatar Morgan Funtowicz
Browse files

Adressing issue in varargs handling for question answering.

parent 46ccbb42
......@@ -293,8 +293,11 @@ class QuestionAnsweringPipeline(Pipeline):
def __call__(self, *args, **kwargs):
# Position args, handling is sensibly the same as X and data, so forwarding to avoid duplicating
if args is not None and len(args) > 1:
kwargs['X'] = args
if args is not None and len(args) > 0:
if len(args) == 1:
kwargs['X'] = args[0]
else:
kwargs['X'] = list(args)
# Generic compatibility with sklearn and Keras
# Batched data
......
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