|
from dataclasses import dataclass |
|
from huggingface_hub import HfApi |
|
from src.backend.manage_requests import EvalRequest |
|
|
|
|
|
@dataclass |
|
class ModelMetadata: |
|
likes: int = 0 |
|
size: int = 15 |
|
|
|
|
|
def sort_models_by_priority(api: HfApi, models: list[EvalRequest]) -> list[EvalRequest]: |
|
private_models = [model for model in models if model.private] |
|
public_models = [model for model in models if not model.private] |
|
|
|
return sort_by_submit_date(private_models) + sort_by_submit_date(public_models) |
|
|
|
|
|
def sort_by_submit_date(eval_requests: list[EvalRequest]) -> list[EvalRequest]: |
|
return sorted(eval_requests, key=lambda x: x.submitted_time, reverse=False) |
|
|
|
|
|
def sort_by_size(eval_requests: list[EvalRequest]) -> list[EvalRequest]: |
|
return sorted(eval_requests, key=lambda x: x.params, reverse=False) |
|
|
|
|
|
def sort_by_likes(eval_requests: list[EvalRequest]) -> list[EvalRequest]: |
|
return sorted(eval_requests, key=lambda x: x.likes, reverse=False) |
|
|
