added torch-specific requirements.txt

wolearyc · Aug 22, 2024 · 9792cba · 9792cba
1 parent 8d8af1b
commit 9792cba
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -26,6 +26,7 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install -r requirements.txt
+        pip install -r torch_extra_requirements.txt
     - name: Analyzing the code with pylint
       run: |
         pylint $(git ls-files '*.py')

diff --git a/requirements.txt b/requirements.txt
@@ -8,5 +8,3 @@ setuptools
 spglib
 tabulate
 torch
-torch-geometric
-torch-sparse
diff --git a/torch_extra_requirements.txt b/torch_extra_requirements.txt
@@ -0,0 +1,4 @@
+# We include these in an extra file, as torch must be pre-installed to install
+# torch-sparse
+torch-geometric
+torch-sparse