diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 837f2ddcc875f33b4403d3e12edaf1ddd1b24bcd..99a31ad515deea8643ce07c0ab5570691308e435 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -26,6 +26,9 @@ jobs: - '3.9' - '3.10' - '3.11' + torch-version: + - '1.7.1' + - 'latest' steps: - name: Get system version for Linux @@ -94,6 +97,15 @@ jobs: pip install -r requirements.txt pip install -r requirements-build.txt pip install -r requirements-test.txt + - name: Install Torch + shell: bash + if: ${{ matrix.torch-version != 'latest' }} + run: | + echo "torch==${{ matrix.torch-version }}; python_version < '3.11'" > requirements-torch.txt + pip install -r requirements-torch.txt + - name: Install Extra Test Requirements + shell: bash + run: | pip install -r requirements-test-extra.txt - name: Test the basic environment shell: bash