Move codelab colab example to using Python 3.9.

PiperOrigin-RevId: 703084261
Change-Id: I2b822f0b27cee8f83b80f552a70e0e5e9dfd9c69
GitOrigin-RevId: 855879c10dfd4a30b612e180695c6be21d8f005d

colab_codelab.ipynb

@@ -57,9 +57,8 @@
         "id": "OvCRFe9_CZ7c"
       },
       "source": [
-        "Since Colab doesn't support Python 3.9, we'll use an older version of XManager that supports Python 3.7.\n",
-        "\n",
-        "You may get some errors during the installation, but they can be ignored for now. For this codelab to work, you must **restart the runtime** after running the cell below for the first time. Avoid using Run All because async functions won't be run properly."
+        "Remember to restart the runtime after running the cell below for the first time. ",
+        "Avoid using run all, as this colab has async functions that will not work correctly."
       ]
     },
     {
@@ -70,9 +69,8 @@
       },
       "outputs": [],
       "source": [
-        "!git clone https://github.com/deepmind/xmanager xmanager_repo\n",
-        "# The latest supported version of Python in Colab is 3.7. Use an older version of XManager that still supports Python 3.7.\n",
-        "!cd xmanager_repo \u0026\u0026 git reset --hard a9ff44972d0a9e18d8a5e85c8ca956903636fad1 \u0026\u0026 pip install ."
+        "!pip install git+https://github.com/deepmind/xmanager\n",
+        "!git clone https://github.com/deepmind/xmanager xmanager_repo"
       ]
     },
     {