Install tf_keras in colabs since it is not in the dependencies yet.

We'll remove it in the next release. PiperOrigin-RevId: 592925209 Change-Id: I50737dbc754291e2e183bee34ee9dad38f562754
tensorflow · Dec 21, 2023 · 27b851f · 27b851f
1 parent c58ea0c
commit 27b851f
Show file tree

Hide file tree

Showing 13 changed files with 26 additions and 13 deletions.
diff --git a/docs/tutorials/10_checkpointer_policysaver_tutorial.ipynb b/docs/tutorials/10_checkpointer_policysaver_tutorial.ipynb
@@ -108,7 +108,8 @@
         "!pip install pyglet\n",
         "!pip install 'imageio==2.4.0'\n",
         "!pip install 'xvfbwrapper==0.2.9'\n",
-        "!pip install tf-agents[reverb]"
+        "!pip install tf-agents[reverb]\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/1_dqn_tutorial.ipynb b/docs/tutorials/1_dqn_tutorial.ipynb
@@ -117,7 +117,8 @@
         "!pip install 'imageio==2.4.0'\n",
         "!pip install pyvirtualdisplay\n",
         "!pip install tf-agents[reverb]\n",
-        "!pip install pyglet"
+        "!pip install pyglet\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/2_environments_tutorial.ipynb b/docs/tutorials/2_environments_tutorial.ipynb
@@ -109,7 +109,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tf-agents[reverb]\n"
+        "!pip install tf-agents[reverb]\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/3_policies_tutorial.ipynb b/docs/tutorials/3_policies_tutorial.ipynb
@@ -111,7 +111,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tf-agents"
+        "!pip install tf-agents\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/4_drivers_tutorial.ipynb b/docs/tutorials/4_drivers_tutorial.ipynb
@@ -102,7 +102,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tf-agents\n"
+        "!pip install tf-agents\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/5_replay_buffers_tutorial.ipynb b/docs/tutorials/5_replay_buffers_tutorial.ipynb
@@ -100,7 +100,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tf-agents\n"
+        "!pip install tf-agents\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/6_reinforce_tutorial.ipynb b/docs/tutorials/6_reinforce_tutorial.ipynb
@@ -114,7 +114,8 @@
         "!pip install 'imageio==2.4.0'\n",
         "!pip install pyvirtualdisplay\n",
         "!pip install tf-agents[reverb]\n",
-        "!pip install pyglet xvfbwrapper\n"
+        "!pip install pyglet xvfbwrapper\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/7_SAC_minitaur_tutorial.ipynb b/docs/tutorials/7_SAC_minitaur_tutorial.ipynb
@@ -109,7 +109,8 @@
         "!pip install 'imageio==2.4.0'\n",
         "!pip install matplotlib\n",
         "!pip install tf-agents[reverb]\n",
-        "!pip install pybullet"
+        "!pip install pybullet\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/8_networks_tutorial.ipynb b/docs/tutorials/8_networks_tutorial.ipynb
@@ -114,7 +114,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tf-agents"
+        "!pip install tf-agents\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/9_c51_tutorial.ipynb b/docs/tutorials/9_c51_tutorial.ipynb
@@ -114,7 +114,8 @@
         "!pip install 'imageio==2.4.0'\n",
         "!pip install pyvirtualdisplay\n",
         "!pip install tf-agents\n",
-        "!pip install pyglet"
+        "!pip install pyglet\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/bandits_tutorial.ipynb b/docs/tutorials/bandits_tutorial.ipynb
@@ -95,7 +95,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tf-agents"
+        "!pip install tf-agents\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/per_arm_bandits_tutorial.ipynb b/docs/tutorials/per_arm_bandits_tutorial.ipynb
@@ -153,7 +153,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tf-agents"
+        "!pip install tf-agents\n",
+        "!pip install tf-keras"
       ]
     },
     {

diff --git a/docs/tutorials/ranking_tutorial.ipynb b/docs/tutorials/ranking_tutorial.ipynb
@@ -86,7 +86,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tf-agents[reverb]"
+        "!pip install tf-agents[reverb]\n",
+        "!pip install tf-keras"
       ]
     },
     {