Use shutil to zip and download colab outputs

PiperOrigin-RevId: 501595938 Change-Id: Ib586090d13279eca1371f0e0b046379b728a3e87
google-deepmind · Jan 12, 2023 · 0d9a24b · 0d9a24b
1 parent a394167
commit 0d9a24b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/notebooks/AlphaFold.ipynb b/notebooks/AlphaFold.ipynb
@@ -342,6 +342,7 @@
         "from concurrent import futures\n",
         "import json\n",
         "import random\n",
+        "import shutil\n",
         "\n",
         "from urllib import request\n",
         "from google.colab import files\n",
@@ -773,7 +774,7 @@
         "    f.write(pae_data)\n",
         "\n",
         "# --- Download the predictions ---\n",
-        "!zip -q -r {output_dir}.zip {output_dir}\n",
+        "shutil.make_archive(base_name='prediction', format='zip', root_dir=output_dir)\n",
         "files.download(f'{output_dir}.zip')"
       ]
     },