From 48d8e1aa41a65d398dfcac4998fd4c02075854af Mon Sep 17 00:00:00 2001
From: tboquet <thomas.boquet@hec.ca>
Date: Tue, 19 Sep 2023 10:01:52 -0400
Subject: [PATCH] fix: add param to constructor

---
 src/tasknet/utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/tasknet/utils.py b/src/tasknet/utils.py
index f6a8214..060e708 100755
--- a/src/tasknet/utils.py
+++ b/src/tasknet/utils.py
@@ -214,6 +214,9 @@ def load_pipeline(
             ] += adapter.Z[task_index]
 
     pipe = TextClassificationPipeline(
-        model=model, tokenizer=tokenizer, device=device, return_all_scores=True
+        model=model,
+        tokenizer=tokenizer,
+        device=device,
+        return_all_scores=return_all_scores,
     )
     return pipe