From 5db18933db3bc010150aff5f0968d229cd0ef61e Mon Sep 17 00:00:00 2001
From: Muhammed Fatih BALIN <m.f.balin@gmail.com>
Date: Wed, 28 Feb 2024 00:48:32 -0500
Subject: [PATCH] [GraphBolt][Tutorial] Fix the typo in the tutorial and colab
 instance type. (#7163)

---
 notebooks/stochastic_training/link_prediction.ipynb | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/notebooks/stochastic_training/link_prediction.ipynb b/notebooks/stochastic_training/link_prediction.ipynb
index 1b7d7b9036..1a1dde0907 100644
--- a/notebooks/stochastic_training/link_prediction.ipynb
+++ b/notebooks/stochastic_training/link_prediction.ipynb
@@ -43,7 +43,7 @@
         "os.environ['TORCH'] = torch.__version__\n",
         "os.environ['DGLBACKEND'] = \"pytorch\"\n",
         "\n",
-        "# Install the CPU version. If you want to install CUDA version, please\n",
+        "# Install the CUDA version. If you want to install CPU version, please\n",
         "# refer to https://www.dgl.ai/pages/start.html.\n",
         "device = torch.device(\"cuda\")\n",
         "!pip install --pre dgl -f https://data.dgl.ai/wheels-test/cu121/repo.html\n",
@@ -348,7 +348,8 @@
   "metadata": {
     "colab": {
       "private_outputs": true,
-      "provenance": []
+      "provenance": [],
+      "gpuType": "T4"
     },
     "kernelspec": {
       "display_name": "Python 3",
@@ -365,8 +366,9 @@
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
       "version": "3.10.12"
-    }
+    },
+    "accelerator": "GPU"
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}
\ No newline at end of file