update

src/backend/manage_requests.py

@@ -5,6 +5,8 @@ from typing import Optional
 
 from huggingface_hub import HfApi, snapshot_download
 
+from src.utils import my_snapshot_download
+
 
 @dataclass
 class EvalRequest:
@@ -67,7 +69,7 @@ def get_eval_requests(job_status: list, local_dir: str, hf_repo: str) -> list[Ev
     Returns:
         `list[EvalRequest]`: a list of model info dicts.
     """
-    snapshot_download(repo_id=hf_repo, revision="main", local_dir=local_dir, repo_type="dataset", max_workers=60)
+    my_snapshot_download(repo_id=hf_repo, revision="main", local_dir=local_dir, repo_type="dataset", max_workers=60)
     json_files = glob.glob(f"{local_dir}/**/*.json", recursive=True)
 
     eval_requests = []
@@ -91,11 +93,9 @@ def get_eval_requests(job_status: list, local_dir: str, hf_repo: str) -> list[Ev
 def check_completed_evals(api: HfApi, hf_repo: str, local_dir: str, checked_status: str, completed_status: str,
                           failed_status: str, hf_repo_results: str, local_dir_results: str):
     """Checks if the currently running evals are completed, if yes, update their status on the hub."""
-
-    from src.utils import my_snapshot_download
     my_snapshot_download(repo_id=hf_repo_results, revision="main", local_dir=local_dir_results, repo_type="dataset", max_workers=60)
 
-    running_evals = get_eval_requests(checked_status, hf_repo=hf_repo, local_dir=local_dir)
+    running_evals = get_eval_requests([checked_status], hf_repo=hf_repo, local_dir=local_dir)
 
     for eval_request in running_evals:
         model = eval_request.model