[TF-DF] Fix uplift colab

PiperOrigin-RevId: 730882503

documentation/tutorials/uplift_colab.ipynb

@@ -82,7 +82,9 @@
         "\n",
         "Install TF-DF by running the following cell.\n",
         "\n",
-        "[Wurlitzer](https://pypi.org/project/wurlitzer/) is needed to display the detailed training logs in Colabs (when using `verbose=2` in the model constructor)."
+        "[Wurlitzer](https://pypi.org/project/wurlitzer/) is needed to display the detailed training logs in Colabs (when using `verbose=2` in the model constructor).\n",
+        "\n",
+        "Tensorflow Datasets is needed download the dataset used in this tutorial."
       ]
     },
     {
@@ -93,7 +95,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tensorflow_decision_forests wurlitzer"
+        "pip install tensorflow_decision_forests wurlitzer tensorflow-datasets"
       ]
     },
     {
@@ -231,18 +233,6 @@
         "Read more about dataset [in its documentation]( https://blog.minethatdata.com/2008/03/minethatdata-e-mail-analytics-and-data.html). This tutorial uses the dataset as curated by  [TensorFlow Datasets](https://www.tensorflow.org/datasets/catalog/hillstrom)."
       ]
     },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "IKkNy8xedOb7"
-      },
-      "outputs": [],
-      "source": [
-        "# Install the TensorFlow Datasets package\n",
-        "!pip install tensorflow-datasets -U --quiet"
-      ]
-    },
     {
       "cell_type": "code",
       "execution_count": null,