diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 1da8f05b..43e560dc 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -30,6 +30,7 @@ jobs: GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} run: | python -m pip install --upgrade pip + pip install setuptools wheel pip install torch==${TORCH} -f https://download.pytorch.org/whl/cpu/torch_stable.html pip install --upgrade-strategy only-if-needed . - name: Install pytest diff --git a/.github/workflows/tests_develop.yml b/.github/workflows/tests_develop.yml index 0f10775e..bae5795e 100644 --- a/.github/workflows/tests_develop.yml +++ b/.github/workflows/tests_develop.yml @@ -30,6 +30,7 @@ jobs: GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} run: | python -m pip install --upgrade pip + pip install setuptools wheel pip install torch==${TORCH} -f https://download.pytorch.org/whl/cpu/torch_stable.html pip install --upgrade-strategy only-if-needed . - name: Install pytest diff --git a/setup.py b/setup.py index 34146189..e827a58b 100644 --- a/setup.py +++ b/setup.py @@ -29,7 +29,7 @@ "numpy", "ase", "tqdm", - "torch>=1.8,<=1.11,!=1.9.0", # torch.fx added in 1.8 + "torch>=1.8,<=1.12,!=1.9.0", # torch.fx added in 1.8 "e3nn>=0.3.5,<0.5.0", "pyyaml", "contextlib2;python_version<'3.7'", # backport of nullcontext