Update testing.py

testing.py

@@ -2,13 +2,8 @@ from transformers import Pipeline
 import torch
 
 class DeclarativePipeline(Pipeline):
-    def _sanitize_parameters(self, **kwargs):
-        preprocess_kwargs = {}
-        if "answer" in kwargs:
-            preprocess_kwargs["answer"] = kwargs["answer"]
-        if "question" in kwargs:
-            preprocess_kwargs["question"] = kwargs["question"]
-        return preprocess_kwargs, {}, {}
+    def _sanitize_parameters(self, *args):
+        return args, {}, {}
 
     def preprocess(self, inputs, question, answer):
         return self.tokenizer(