From beea0728cf7fd89a1573d98b9dd610d6c6cbc6c8 Mon Sep 17 00:00:00 2001 From: Felipe Alex Hofmann Date: Fri, 4 Dec 2020 18:57:58 -0800 Subject: [PATCH] Changes required torch version --- .github/workflows/tests.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index a71f7cfe..7cd6563a 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -21,7 +21,7 @@ jobs: python-version: ${{ matrix.python-version }} - if: matrix.os == 'windows-latest' name: Install dependencies - Windows - run: pip install 'torch>=1,<2' -f https://download.pytorch.org/whl/torch_stable.html + run: pip install torch==1.7.0+cpu torchvision==0.8.1+cpu -f https://download.pytorch.org/whl/torch_stable.html - name: Install package run: pip install invoke .[dev] - name: invoke lint @@ -58,7 +58,7 @@ jobs: python-version: ${{ matrix.python-version }} - if: matrix.os == 'windows-latest' name: Install dependencies - Windows - run: pip install 'torch>=1,<2' -f https://download.pytorch.org/whl/torch_stable.html + run: pip install torch==1.7.0+cpu torchvision==0.8.1+cpu -f https://download.pytorch.org/whl/torch_stable.html - name: Install package and dependencies run: pip install invoke .[test] - name: invoke pytest