Fixing ruff top import

lightonai · Oct 17, 2024 · a855677 · a855677
1 parent 581a614
commit a855677
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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]