Missing import and pip install of tf-nightly in workshop notebook

PiperOrigin-RevId: 204736577

1 files changed, 21 insertions, 3 deletions

diff --git a/tensorflow/contrib/autograph/examples/notebooks/workshop.ipynb b/tensorflow/contrib/autograph/examples/notebooks/workshop.ipynb
index e8f16b431d..4643656ff4 100644
--- a/tensorflow/contrib/autograph/examples/notebooks/workshop.ipynb
+++ b/tensorflow/contrib/autograph/examples/notebooks/workshop.ipynb
@@ -11,14 +11,32 @@
           }
         },
         "colab_type": "code",
+        "id": "yFn4b8J0CKQV"
+      },
+      "outputs": [],
+      "source": [
+        "pip install -U tf-nightly"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 0,
+      "metadata": {
+        "colab": {
+          "autoexec": {
+            "startup": false,
+            "wait_interval": 0
+          }
+        },
+        "colab_type": "code",
         "id": "qWUV0FYjDSKj"
       },
       "outputs": [],
       "source": [
+        "import os\n",
+        "import matplotlib.pyplot as plt\n",
         "import tensorflow as tf\n",
-        "from tensorflow.contrib import autograph\n",
-        "\n",
-        "import matplotlib.pyplot as plt"
+        "from tensorflow.contrib import autograph"
       ]
     },
     {