From ed1b814008101049e8655c5658f2f4417d0ddbc1 Mon Sep 17 00:00:00 2001
From: Sergio Gomez <sergomez@google.com>
Date: Thu, 23 Jul 2020 12:12:05 +0100
Subject: [PATCH] Fix problem with gsutil command to copy Atari data to local
 folder

PiperOrigin-RevId: 322759837
---
 rl_unplugged/atari_dqn.ipynb | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/rl_unplugged/atari_dqn.ipynb b/rl_unplugged/atari_dqn.ipynb
index 0aa71e5..bd5a54e 100644
--- a/rl_unplugged/atari_dqn.ipynb
+++ b/rl_unplugged/atari_dqn.ipynb
@@ -126,7 +126,10 @@
         "gs_path = 'gs://rl_unplugged/atari'\n",
         "\n",
         "!mkdir -p {tmp_path}/{game}\n",
-        "!gsutil cp {gs_path}/{game}/run_{run}-00000-of-00001 {tmp_path}/{game}"
+        "\n",
+        "src = f'{gs_path}/{game}/run_{run}-00000-of-00100'\n",
+        "dest = f'{tmp_path}/{game}/run_{run}-00000-of-00001'\n",
+        "!gsutil cp {src} {dest}"
       ]
     },
     {
@@ -386,8 +389,8 @@
     "colab": {
       "collapsed_sections": [],
       "last_runtime": {
-        "build_target": "",
-        "kind": "local"
+        "build_target": "//learning/deepmind/dm_python:dm_notebook3",
+        "kind": "private"
       },
       "name": "RL Unplugged: Offline DQN - Atari",
       "provenance": [