From a8556779d2791cdaf3507bad7ab6353ba2ce480d Mon Sep 17 00:00:00 2001 From: Antoine Chaffin Date: Tue, 15 Oct 2024 13:43:37 +0000 Subject: [PATCH] Fixing ruff top import --- pylate/model_card.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pylate/model_card.py b/pylate/model_card.py index ba7e61c..b791ba3 100644 --- a/pylate/model_card.py +++ b/pylate/model_card.py @@ -19,6 +19,7 @@ from huggingface_hub import model_info as get_model_info from huggingface_hub.repocard_data import EvalResult, eval_results_to_model_index from huggingface_hub.utils import yaml_dump +from sentence_transformers import SentenceTransformerModelCardData from sentence_transformers import __version__ as sentence_transformers_version from sentence_transformers.models import StaticEmbedding, Transformer from sentence_transformers.training_args import SentenceTransformerTrainingArguments @@ -47,9 +48,6 @@ from sentence_transformers.trainer import SentenceTransformerTrainer -from sentence_transformers import SentenceTransformerModelCardData - - class ModelCardCallback(TrainerCallback): def __init__( self, trainer: SentenceTransformerTrainer, default_args_dict: dict[str, Any]