Nathan Habib commited on
Commit
26286b2
1 Parent(s): f3cda22

add safefail for when we cannot download datasets, will simply restart the space

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -91,8 +91,15 @@ BENCHMARK_COLS = [
91
  ]
92
  ]
93
 
94
- snapshot_download(repo_id=QUEUE_REPO, local_dir=EVAL_REQUESTS_PATH, repo_type="dataset", tqdm_class=None)
95
- snapshot_download(repo_id=RESULTS_REPO, local_dir=EVAL_RESULTS_PATH, repo_type="dataset", tqdm_class=None)
 
 
 
 
 
 
 
96
  requested_models, users_to_submission_dates = get_all_requested_models(EVAL_REQUESTS_PATH)
97
 
98
  original_df = get_leaderboard_df(EVAL_RESULTS_PATH, COLS, BENCHMARK_COLS)
 
91
  ]
92
  ]
93
 
94
+ try:
95
+ snapshot_download(repo_id=QUEUE_REPO, local_dir=EVAL_REQUESTS_PATH, repo_type="dataset", tqdm_class=None, etag_timeout=30)
96
+ except Exception:
97
+ restart_space()
98
+ try:
99
+ snapshot_download(repo_id=RESULTS_REPO, local_dir=EVAL_RESULTS_PATH, repo_type="dataset", tqdm_class=None, etag_timeout=30)
100
+ except Exception:
101
+ restart_space()
102
+
103
  requested_models, users_to_submission_dates = get_all_requested_models(EVAL_REQUESTS_PATH)
104
 
105
  original_df = get_leaderboard_df(EVAL_RESULTS_PATH, COLS, BENCHMARK_COLS)