From 68154490188bd061f835dba69ca5fa0df7ff4c72 Mon Sep 17 00:00:00 2001
From: Matteo Bettini <mb2389@cl.cam.ac.uk>
Date: Mon, 27 Nov 2023 14:18:23 +0000
Subject: [PATCH] setup

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 90afac45..a4e50f91 100644
--- a/setup.py
+++ b/setup.py
@@ -36,6 +36,7 @@ def package_files(directory):
     extras_require={
         "vmas": ["vmas>=1.2.10"],
         "pettingzoo": ["pettingzoo[all]>=1.24.1"],
+        "gnn": ["torch_geometric"],
     },
     packages=find_packages(),
     include_package_data=True,