All the modes work the same way

Add pip install in the beginning

Author: orazve

examples/kge-predict-transe-pyg-train.ipynb

@@ -3,7 +3,17 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%pip install graphdatascience torch torch_geometric"
+   ],
+   "metadata": {
+    "collapsed": false
+   }
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
    "outputs": [],
    "source": [
     "from graphdatascience import GraphDataScience\n",
@@ -12,7 +22,10 @@
     "from torch_geometric.data import Data, download_url\n",
     "from torch_geometric.nn import TransE\n",
     "import collections"
-   ]
+   ],
+   "metadata": {
+    "collapsed": false
+   }
   },
   {
    "cell_type": "code",
@@ -300,7 +313,7 @@
     "                \"EMBEDDING\": model.node_emb.weight[i].tolist()\n",
     "            },\n",
     "        )\n",
-    "    print(f\"Node embeddings uploading has finished\")"
+    "print(f\"Node embeddings uploading has been finished\")"
    ]
   },
   {