Replace gsutil commands with gcloud storage in Dopamine Colab notebooks.

PiperOrigin-RevId: 888442415

Author: Dopamine Team

dopamine/colab/agent_visualizer.ipynb

@@ -68,9 +68,9 @@
       "outputs": [],
       "source": [
         "# @title Download an example checkpoint (Rainbow on SpaceInvaders)\n",
-        "!gsutil -q -m cp -R gs://download-dopamine-rl/colab/samples/rainbow/SpaceInvaders_v4/checkpoints/tf_ckpt-199.data-00000-of-00001 /tmp\n",
-        "!gsutil -q -m cp -R gs://download-dopamine-rl/colab/samples/rainbow/SpaceInvaders_v4/checkpoints/tf_ckpt-199.index /tmp\n",
-        "!gsutil -q -m cp -R gs://download-dopamine-rl/colab/samples/rainbow/SpaceInvaders_v4/checkpoints/tf_ckpt-199.meta /tmp"
+        "!gcloud storage cp --recursive gs://download-dopamine-rl/colab/samples/rainbow/SpaceInvaders_v4/checkpoints/tf_ckpt-199.data-00000-of-00001 /tmp\n",
+        "!gcloud storage cp --recursive gs://download-dopamine-rl/colab/samples/rainbow/SpaceInvaders_v4/checkpoints/tf_ckpt-199.index /tmp\n",
+        "!gcloud storage cp --recursive gs://download-dopamine-rl/colab/samples/rainbow/SpaceInvaders_v4/checkpoints/tf_ckpt-199.meta /tmp"
       ]
     },
     {

dopamine/colab/jax_agent_visualizer.ipynb

@@ -111,7 +111,7 @@
         "# @title Download a trained checkpoint\n",
         "agent = 'quantile'  # @param ['dqn', 'c51', 'rainbow', 'quantile', 'implicit_quantile']\n",
         "game = 'Breakout'  # @param ['Pong', 'SpaceInvaders', 'Seaquest', 'Qbert', 'Breakout', 'Asterix']\n",
-        "!gsutil -q -m cp -R gs://download-dopamine-rl/jax/{agent}/{game}/1/ckpt.199 /tmp"
+        "!gcloud storage cp --recursive gs://download-dopamine-rl/jax/{agent}/{game}/1/ckpt.199 /tmp"
       ],
       "execution_count": 31,
       "outputs": []

dopamine/colab/load_statistics.ipynb

@@ -88,7 +88,7 @@
         "\n",
         "# For illustrative purposes, we are providing sample logs of the Rainbow agent\n",
         "# trained without sticky actions.\n",
-        "!gsutil -q -m cp -R gs://download-dopamine-rl/colab/* /content/"
+        "!gcloud storage cp --recursive gs://download-dopamine-rl/colab/* /content/"
       ]
     },
     {
@@ -114,7 +114,7 @@
       "source": [
         "# @title Load the baseline data\n",
         "\n",
-        "!gsutil -q -m cp -R gs://download-dopamine-rl/preprocessed-benchmarks/* /content/\n",
+        "!gcloud storage cp --recursive gs://download-dopamine-rl/preprocessed-benchmarks/* /content/\n",
         "experimental_data = colab_utils.load_baselines('/content')"
       ]
     },

dopamine/colab/tensorboard.ipynb

@@ -53,7 +53,7 @@
         "# @title Prepare all necessary files and load TensorBoard extension.\n",
         "%load_ext tensorboard\n",
         "\n",
-        "!gsutil -q -m cp -R gs://download-dopamine-rl/compiled_tb_event_files.tar.gz /content/\n",
+        "!gcloud storage cp --recursive gs://download-dopamine-rl/compiled_tb_event_files.tar.gz /content/\n",
         "!tar -xvzf /content/compiled_tb_event_files.tar.gz"
       ],
       "execution_count": null,