5 retries max

src/leaderboard/populate.py

@@ -16,7 +16,7 @@ def download_result_data():
     os.environ["HF_HUB_ETAG_TIMEOUT"] = "30"
     os.environ["HF_HUB_DOWNLOAD_TIMEOUT"] = "30"
 
-    while True:
+    for i in range(5):
         try:
             snapshot_download(
                 repo_id=RESULTS_REPO_ID,
@@ -27,14 +27,16 @@ def download_result_data():
                 token=TOKEN,
                 allow_patterns=["results/*"],
             )
-            break
+            return
         except Exception as e:
-            if isinstance(e, (ReadTimeout, ConnectionError)):
-                time.sleep(3)
-                print(f"Connection error: {e}. Retrying...")
-                continue
-            else:
-                raise e
+            print(f"Connection error: {e.__class__.__name__} {e}. Retrying in {3 * (i + 1)} seconds...")
+            time.sleep(3 * (i + 1))
+            continue
+
+    raise RuntimeError(
+        f"Failed to download {RESULTS_REPO_ID} after 5 attempts. "
+        "Please check your internet connection or the repository."
+    )
 
 
 def load_results() -> pd.DataFrame: