From aa7bd31986429813f4a87032da2875d01e65d768 Mon Sep 17 00:00:00 2001
From: Federico Berto <berto.federico2@gmail.com>
Date: Tue, 22 Aug 2023 18:43:08 +0900
Subject: [PATCH] [Notebooks] add Torch Geometric dependency to graph encoders

---
 notebooks/tutorials/3-change-encoder.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/notebooks/tutorials/3-change-encoder.ipynb b/notebooks/tutorials/3-change-encoder.ipynb
index 7eb4b639..16c6d507 100644
--- 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"