Tru to fix requirements.

materialsvirtuallab · Aug 13, 2024 · b392449 · b392449
1 parent 57e22a1
commit b392449
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -50,12 +50,12 @@ classifiers = [
 ]
 dependencies = [
     "ase",
-    "dgl>=2.0.0",
+    "dgl<=2.1.0",
     "pymatgen",
-    "lightning",
+    "lightning<=2.4.0",
     "torch<=2.2.1",
     "pydantic",
-    "torchdata",
+    "torchdata<=0.7.1",
     "boto3",
     "numpy<2.0.0"
 ]