Update DTensor docs, lint notebooks

tensorflow · Sep 29, 2023 · 9e06301 · 9e06301
1 parent a3d095e
commit 9e06301
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 28 deletions.
diff --git a/site/en/guide/dtensor_overview.ipynb b/site/en/guide/dtensor_overview.ipynb
@@ -89,29 +89,7 @@
         "\n",
         "DTensor (`tf.experimental.dtensor`) has been part of TensorFlow since the 2.9.0 release.\n",
         "\n",
-        "First, install or upgrade TensorFlow:"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "OKaPw8vwwZAC"
-      },
-      "outputs": [],
-      "source": [
-        "!pip install --quiet --upgrade tensorflow"
-      ]
-    },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "O3pG29uZIWYO"
-      },
-      "source": [
-        "Then, import `tensorflow` and `dtensor`, and configure TensorFlow to use 6 virtual CPUs.\n",
-        "\n",
-        "Even though this example uses virtual CPUs, DTensor works the same way on CPU, GPU or TPU devices."
+        "Begin by importing TensorFlow, `dtensor`, and configure TensorFlow to use 6 virtual CPUs. Even though this example uses virtual CPUs, DTensor works the same way on CPU, GPU or TPU devices."
       ]
     },
     {

diff --git a/site/en/tutorials/distribute/dtensor_keras_tutorial.ipynb b/site/en/tutorials/distribute/dtensor_keras_tutorial.ipynb
@@ -91,7 +91,7 @@
         "\n",
         "DTensor (`tf.experimental.dtensor`) has been part of TensorFlow since the 2.9.0 release.\n",
         "\n",
-        "First, install or upgrade TensorFlow and TensorFlow Datasets:"
+        "First, install or upgrade TensorFlow Datasets:"
       ]
     },
     {
@@ -102,7 +102,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install --quiet --upgrade tensorflow tensorflow-datasets"
+        "!pip install --quiet --upgrade tensorflow-datasets"
       ]
     },
     {
@@ -111,7 +111,7 @@
         "id": "VttBMZngDx8x"
       },
       "source": [
-        "Next, import `tensorflow` and `dtensor`, and configure TensorFlow to use 8 virtual CPUs.\n",
+        "Next, import TensorFlow and `dtensor`, and configure TensorFlow to use 8 virtual CPUs.\n",
         "\n",
         "Even though this example uses virtual CPUs, DTensor works the same way on CPU, GPU or TPU devices."
       ]

diff --git a/site/en/tutorials/distribute/dtensor_ml_tutorial.ipynb b/site/en/tutorials/distribute/dtensor_ml_tutorial.ipynb
@@ -98,7 +98,7 @@
         "\n",
         "DTensor (`tf.experimental.dtensor`) has been part of TensorFlow since the 2.9.0 release.\n",
         "\n",
-        "First, install or upgrade TensorFlow and TensorFlow Datasets:"
+        "First, install or upgrade TensorFlow Datasets:"
       ]
     },
     {
@@ -109,7 +109,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install --quiet --upgrade tensorflow tensorflow-datasets"
+        "!pip install --quiet --upgrade tensorflow-datasets"
       ]
     },
     {