From 61408431912abc942f7d0c319b2fd9d1624ab93b Mon Sep 17 00:00:00 2001
From: Eric Sivonxay <esivonxay@lbl.gov>
Date: Thu, 26 Oct 2023 21:58:21 -0700
Subject: [PATCH] Limit pytorch geometric version

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

diff --git a/setup.py b/setup.py
index bc08625..024dc9c 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
           "ray",
           "wandb",
           "gpuparallel",
-          "torch-geometric",
+          "torch-geometric<=2.3.1",
           "h5py"
       ]
       )