Fix minor bug in transformers ext

superduper-io · Nov 28, 2023 · d673f95 · d673f95
1 parent ea8ec01
commit d673f95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/superduperdb/ext/transformers/model.py b/superduperdb/ext/transformers/model.py
@@ -193,7 +193,7 @@ def compute_metrics(eval_pred):
 
     def _predict(self, X, one: bool = False, **kwargs):
         if self.pipeline is not None:
-            out = self.pipeline(X, **self.preprocess_kwargs, **kwargs)
+            out = self.pipeline(X, **kwargs)
             out = [r['label'] for r in out]
             for i, p in enumerate(out):
                 if re.match(r'^LABEL_[0-9]+', p):