Update download script

fontanf · Apr 15, 2024 · 5605faa · 5605faa
1 parent 2671088
commit 5605faa
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/scripts/download_data.py b/scripts/download_data.py
@@ -1,12 +1,20 @@
 import gdown
 import os
 import pathlib
+import time
 
 
 def download(id):
-    gdown.download(id=id, output="data.7z")
-    os.system("7z x data.7z -odata")
-    pathlib.Path("data.7z").unlink()
+    for _ in range(3):
+        try:
+            gdown.download(id=id, output="data.7z")
+            os.system("7z x data.7z -odata")
+            pathlib.Path("data.7z").unlink()
+        except:
+            time.sleep(10)
+            continue
+        return
+    sys.exit(1)
 
 
 download("1_sbLgEoWCFo4gEsg9BE8eG4De4f8fkF0")