EmbodiedAgentInterface/backend

1import re2from dataclasses import dataclass3 4from huggingface_hub import HfApi5 6from src.backend.manage_requests import EvalRequest7 8 9@dataclass10class ModelMetadata:11    likes: int = 012    size: int = 1513 14 15def sort_models_by_priority(api: HfApi, models: list[EvalRequest]) -> list[EvalRequest]:16    private_models = [model for model in models if model.private]17    public_models = [model for model in models if not model.private]18 19    return sort_by_submit_date(private_models) + sort_by_submit_date(public_models)20 21def sort_by_submit_date(eval_requests: list[EvalRequest]) -> list[EvalRequest]:22    return sorted(eval_requests, key=lambda x: x.submitted_time, reverse=False)23 24def sort_by_size(eval_requests: list[EvalRequest]) -> list[EvalRequest]:25    return sorted(eval_requests, key=lambda x: x.params, reverse=False)26 27def sort_by_likes(eval_requests: list[EvalRequest]) -> list[EvalRequest]:28    return sorted(eval_requests, key=lambda x: x.likes, reverse=False)