Pin torch-scatter version

wengroup · Dec 11, 2023 · e56d554 · e56d554
1 parent 01c37da
commit e56d554
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,7 @@ dependencies = [
     "pytorch-lightning>=2.0.0",
     "torchmetrics>=0.10.0, <1.0.0",
     "torch_geometric>=2.3.0",
-    "torch_scatter",
+    "torch_scatter==2.1.1",
     "e3nn",
     "ase",
     "pymatgen",