setup

facebookresearch · Nov 27, 2023 · 6815449 · 6815449
1 parent 71e821f
commit 6815449
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,