Merge pull request #2726 from tomaarsen/fix/no_evaluator

[`fix`] Fix edge case with evaluator being None
UKPLab · Jun 7, 2024 · d9c2b0c · d9c2b0c
2 parents b5e98e1 + 4d3e357
commit d9c2b0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sentence_transformers/trainer.py b/sentence_transformers/trainer.py
@@ -223,7 +223,7 @@ def __init__(
         else:
             self.loss = self.prepare_loss(loss, model)
         # If evaluator is a list, we wrap it in a SequentialEvaluator
-        if not isinstance(evaluator, SentenceEvaluator):
+        if evaluator is not None and not isinstance(evaluator, SentenceEvaluator):
             evaluator = SequentialEvaluator(evaluator)
         self.evaluator = evaluator