Merge branch 'devel'

wengroup · Dec 11, 2023 · 60b83c1 · 60b83c1
2 parents fd0743c + c90a758
commit 60b83c1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
@@ -48,8 +48,9 @@ jobs:
       - name: Install package
         run: |
           python -m pip install --upgrade pip
-          pip install torch
-          pip install -e .[test]
+          pip install torch==2.1.1
+          pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.0+cpu.html
+          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",