diff --git a/pyproject.toml b/pyproject.toml
index 66e617b..5479117 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -56,7 +56,7 @@ dependencies = [
   "einops>=0.6.1",
   "hydra-core>=1.3",
   "torch>=2.2",
-  "torch-geometric>=2.3,<2.5",
+  "torch-geometric>=2.3.1",
 ]
 optional-dependencies.all = [
 ]