File size: 959 Bytes
14e4843
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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)