Upload NLIScorer
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -366,6 +366,7 @@ class NLIScorer(Pipeline):
|
|
| 366 |
if "task_type" in kwargs:
|
| 367 |
preprocess_kwargs["task_type"] = kwargs.get("task_type")
|
| 368 |
postprocess_kwargs["task_type"] = kwargs.get("task_type")
|
|
|
|
| 369 |
return preprocess_kwargs, {}, postprocess_kwargs
|
| 370 |
|
| 371 |
def preprocess(self, inputs, task_type=None):
|
|
|
|
| 366 |
if "task_type" in kwargs:
|
| 367 |
preprocess_kwargs["task_type"] = kwargs.get("task_type")
|
| 368 |
postprocess_kwargs["task_type"] = kwargs.get("task_type")
|
| 369 |
+
postprocess_kwargs["threshold"] = kwargs.get("threshold")
|
| 370 |
return preprocess_kwargs, {}, postprocess_kwargs
|
| 371 |
|
| 372 |
def preprocess(self, inputs, task_type=None):
|