Fix AssertionError in Debug TensorFlow 2 migrated training pipeline guide

PiperOrigin-RevId: 491740481

Author: 8bitmp3

site/en/guide/migrate/migration_debugging.ipynb

@@ -37,7 +37,7 @@
         "id": "77z2OchJTk0l"
       },
       "source": [
-        "# Debug TF2 Migrated Training Pipeline\n",
+        "# Debug a TensorFlow 2 migrated training pipeline\n",
         "\n",
         "<table class=\"tfo-notebook-buttons\" align=\"left\">\n",
         "  <td>\n",
@@ -67,12 +67,12 @@
         "id": "zTwPu-w6M5sz"
       },
       "source": [
-        "This notebook demonstrates how to debug training pipeline when migrating to TF2. It consists of following components:\n",
+        "This notebook demonstrates how to debug a training pipeline when migrating to TensorFlow 2 (TF2). It consists of following components:\n",
         "1. Suggested steps and code samples for debugging training pipeline\n",
         "2. Tools for debugging\n",
         "3. Other related resources\n",
         "\n",
-        "One assumption is you have TF1.x code and trained models for comparison, and you want to build a TF2 model that achieves similar validation accuracy.\n",
+        "One assumption is you have the TensorFlow 1 (TF1.x) code and trained models for comparison, and you want to build a TF2 model that achieves similar validation accuracy.\n",
         "\n",
         "This notebook does **NOT** cover debugging performance issues for training/inference speed or memory usage."
       ]
@@ -151,17 +151,6 @@
         "## Setup"
       ]
     },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "sopP--i7-LaF"
-      },
-      "outputs": [],
-      "source": [
-        "!pip uninstall -y -q tensorflow"
-      ]
-    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -170,9 +159,8 @@
       },
       "outputs": [],
       "source": [
-        "# Install tf-nightly as the DeterministicRandomTestTool is only available in\n",
-        "# Tensorflow 2.8\n",
-        "!pip install -q tf-nightly"
+        "# The `DeterministicRandomTestTool` is only available from Tensorflow 2.8:\n",
+        "!pip install -q \"tensorflow==2.9.*\""
       ]
     },
     {