Skip to content

Commit

Permalink
Fixing ruff top import
Browse files Browse the repository at this point in the history
  • Loading branch information
NohTow committed Oct 15, 2024
1 parent 997fc0b commit 50954f8
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions pylate/model_card.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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]
Expand Down

0 comments on commit 50954f8

Please sign in to comment.