diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml index a39bc75..4568abe 100644 --- 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 index ae8bebe..d1ca721 100644 --- 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 new file mode 100644 index 0000000..36930ab --- /dev/null +++ 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