Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
pminervini
commited on
Commit
•
5a0f81d
1
Parent(s):
c0db8b3
update
Browse files
src/backend/manage_requests.py
CHANGED
@@ -91,7 +91,9 @@ def get_eval_requests(job_status: list, local_dir: str, hf_repo: str) -> list[Ev
|
|
91 |
def check_completed_evals(api: HfApi, hf_repo: str, local_dir: str, checked_status: str, completed_status: str,
|
92 |
failed_status: str, hf_repo_results: str, local_dir_results: str):
|
93 |
"""Checks if the currently running evals are completed, if yes, update their status on the hub."""
|
94 |
-
|
|
|
|
|
95 |
|
96 |
running_evals = get_eval_requests(checked_status, hf_repo=hf_repo, local_dir=local_dir)
|
97 |
|
|
|
91 |
def check_completed_evals(api: HfApi, hf_repo: str, local_dir: str, checked_status: str, completed_status: str,
|
92 |
failed_status: str, hf_repo_results: str, local_dir_results: str):
|
93 |
"""Checks if the currently running evals are completed, if yes, update their status on the hub."""
|
94 |
+
|
95 |
+
from src.utils import my_snapshot_download
|
96 |
+
my_snapshot_download(repo_id=hf_repo_results, revision="main", local_dir=local_dir_results, repo_type="dataset", max_workers=60)
|
97 |
|
98 |
running_evals = get_eval_requests(checked_status, hf_repo=hf_repo, local_dir=local_dir)
|
99 |
|