pminervini commited on
Commit
5a0f81d
1 Parent(s): c0db8b3
Files changed (1) hide show
  1. src/backend/manage_requests.py +3 -1
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
- snapshot_download(repo_id=hf_repo_results, revision="main", local_dir=local_dir_results, repo_type="dataset", max_workers=60)
 
 
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