[Notebooks] add Torch Geometric dependency to graph encoders

ai4co · Aug 22, 2023 · aa7bd31 · aa7bd31
1 parent 1b6f027
commit aa7bd31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/tutorials/3-change-encoder.ipynb b/notebooks/tutorials/3-change-encoder.ipynb
@@ -28,7 +28,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# !pip install rl4co\n",
+    "# !pip install rl4co[graph] # include torch-geometric\n",
     "\n",
     "## NOTE: to install latest version from Github (may be unstable) install from source instead:\n",
     "# !pip install git+https://github.com/kaist-silab/rl4co.git"