Fix the file loading code in OSS colab.

PiperOrigin-RevId: 608751535 Change-Id: I4dc4d83e0d8936c33198936824a7c62aa2b73068
google-research · Feb 20, 2024 · a5af8c4 · a5af8c4
1 parent b2c15ba
commit a5af8c4
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/mf_dpftrl_matrices/privacy_accounting.ipynb b/mf_dpftrl_matrices/privacy_accounting.ipynb
@@ -106,12 +106,16 @@
       },
       "outputs": [],
       "source": [
-        "b1_file = tf.keras.utils.get_file(b1_url)\n",
-        "c1_file = tf.keras.utils.get_file(c1_url)\n",
-        "lr1_file = tf.keras.utils.get_file(lr1_url)\n",
-        "\n",
-        "b2_file = tf.keras.utils.get_file(b2_url)\n",
-        "c2_file = tf.keras.utils.get_file(c2_url)"
+        "b1_file = tf.keras.utils.get_file(fname='b1', origin=b1_url)\n",
+        "c1_file = tf.keras.utils.get_file(fname='c1', origin=c1_url)\n",
+        "lr1_file = tf.keras.utils.get_file(fname='lr1', origin=lr1_url)\n",
+        "\n",
+        "b2_file = tf.keras.utils.get_file(fname='b2', origin=b2_url)\n",
+        "c2_file = tf.keras.utils.get_file(fname='c2', origin=c2_url)\n",
+        "print(\n",
+        "    f'Local data location:\\n {b1_file}\\n {c1_file}\\n {lr1_file}\\n {b2_file}\\n'\n",
+        "    f' {c2_file}'\n",
+        ")"
       ]
     },
     {