diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index fbf5dd0..919da80 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -32,7 +32,7 @@ jobs: max-parallel: 2 matrix: python-version: ['3.9'] - torch-version: ['--upgrade'] + torch-version: [2.1.1] os: [ubuntu-latest] # only run ubuntu for now because the other ones fail for no reason, macos-latest, windows-latest] # Steps represent a sequence of tasks that will be executed as part of the job