Merge branch 'devel' of github.com:wengroup/matten into devel

wengroup · Dec 11, 2023 · 4fc5254 · 4fc5254
2 parents ca7c51d + 2d470d9
commit 4fc5254
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
@@ -49,7 +49,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install torch
-          pip install -e .[test]
+          pip install -e .[strict,test]
 
       - name: Test
         run: pytest
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,6 +8,7 @@ version = "0.0.1"
 readme = "README.md"
 requires-python = ">=3.8, <3.11"
 dependencies = [
+    "torch>=2.0.0",
     "pytorch-lightning>=2.0.0",
     "torchmetrics>=0.10.0, <1.0.0",
     "torch_geometric>=2.3.0",
@@ -21,6 +22,14 @@ dependencies = [
 
 [project.optional-dependencies]
 test = ["pytest"]
+strict = [
+    "pytorch-lightning==2.0.0",
+    "torchmetrics==0.10.0",
+    "torch_geometric==2.3.0",
+    "torch_scatter==2.1.2",
+    "e3nn==0.5.1",
+]
+
 
 [tool.black]
 line-length = 88