add pyg pins

recursionpharma · Oct 26, 2023 · 4019030 · 4019030
1 parent e36610d
commit 4019030
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -57,10 +57,10 @@ requires-python = ">=3.8,<3.10"
 dynamic = ["version"]
 dependencies = [
     "torch==1.13.1",
-    "torch-geometric",
-    "torch-scatter",
-    "torch-sparse",
-    "torch-cluster",
+    "torch-geometric==2.3.1",  # Pinning until we adapt the code to newer versions
+    "torch-scatter==2.1.1",
+    "torch-sparse==0.6.17",
+    "torch-cluster==1.6.1",
     "rdkit",
     "tables",
     "scipy",