Limit pytorch geometric version

BlauGroup · Oct 27, 2023 · 6140843 · 6140843
1 parent 3e1b9f6
commit 6140843
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -23,7 +23,7 @@
           "ray",
           "wandb",
           "gpuparallel",
-          "torch-geometric",
+          "torch-geometric<=2.3.1",
           "h5py"
       ]
       )