From b259dd3ef1dfb72ed76ad2b5e9c25a5bcf2787da Mon Sep 17 00:00:00 2001 From: Brian Lawney Date: Thu, 15 May 2025 15:41:47 -0400 Subject: [PATCH] Address deprecated ModelCardCallback from sentence_transformers --- src/setfit/trainer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/setfit/trainer.py b/src/setfit/trainer.py index 47a063ab..f51f26a7 100644 --- a/src/setfit/trainer.py +++ b/src/setfit/trainer.py @@ -7,7 +7,8 @@ from packaging.version import parse as parse_version from sentence_transformers import SentenceTransformerTrainer, losses from sentence_transformers.losses.BatchHardTripletLoss import BatchHardTripletLossDistanceFunction -from sentence_transformers.model_card import ModelCardCallback as STModelCardCallback +from sentence_transformers.model_card import ModelCardCallback as DeprecatedSTModelCardCallback +from sentence_transformers.model_card import SentenceTransformerModelCardCallback from sentence_transformers.training_args import BatchSamplers, SentenceTransformerTrainingArguments from sklearn.preprocessing import LabelEncoder from torch import nn @@ -58,7 +59,7 @@ def __init__( if isinstance(callback, CodeCarbonCallback): self.setfit_model.model_card_data.code_carbon_callback = callback - if isinstance(callback, STModelCardCallback): + if isinstance(callback, (DeprecatedSTModelCardCallback, SentenceTransformerModelCardCallback)): self.remove_callback(callback) if is_in_notebook():