From ed1ea7c5373391d9aeb44e939542928a05aa40ee Mon Sep 17 00:00:00 2001
From: FilippoOlivo <filippo@filippoolivo.com>
Date: Mon, 25 Nov 2024 17:47:13 +0100
Subject: [PATCH] Add pytorch_geometric in setup.py file

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 04f7b2fc..ea8d05ad 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,8 @@
 KEYWORDS = 'physics-informed neural-network'
 
 REQUIRED = [
-    'numpy', 'matplotlib', 'torch', 'lightning', 'torch_geometric', 'torch-cluster'
+    'numpy', 'matplotlib', 'torch', 'lightning', 'pytorch_lightning',
+    'torch_geometric', 'torch-cluster'
 ]
 
 EXTRAS = {