Updated pytorch version for CI (#7)

MrParosk · Apr 28, 2024 · 66434b6 · 66434b6
1 parent 3be6f1b
commit 66434b6
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/.github/workflows/master.yml b/.github/workflows/master.yml
@@ -14,8 +14,8 @@ jobs:
       - name: Install dependencies
         run: |
           sudo apt update && sudo apt-get install -y wget unzip cmake libgtest-dev clang-format
-      - name: Fetch libtorch # TODO: using a older version since 2.2.2 had some changes to how caffe2::mkl was included. Need to fix
-        run: wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.1.2%2Bcpu.zip && unzip libtorch-cxx11-abi-shared-with-deps-2.1.2+cpu.zip
+      - name: Fetch libtorch
+        run: wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Bcpu.zip && unzip libtorch-cxx11-abi-shared-with-deps-2.3.0+cpu.zip
       - name: Run format checking
         run: bash scripts/run_format.sh -c
       - name: Run tests
@@ -46,21 +46,23 @@ jobs:
         mypy .
     - name: Install pytorch
       run: |
-        python -m pip install torch==2.2.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
+        python -m pip install torch==2.3.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
     - name: Code coverage
       run: |
         python -m pip install --no-build-isolation -e .
         pytest --cov=pt_soft_nms --cov-report=xml
     - name: Upload coverage to Codecov
-      uses: codecov/codecov-action@v1
+      uses: codecov/codecov-action@v4
+      with:
+        token: ${{ secrets.CODECOV_TOKEN }}
 
   test-python:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
         python-version: ['3.9', '3.10', '3.11']
         os: [ubuntu-latest]
-        pytorch-version: ['1.13.1', '2.2.2']
+        pytorch-version: ['1.13.1', '2.3.0']
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python
@@ -94,7 +96,7 @@ jobs:
         python -m pip install -r requirements-dev.txt
     - name: Install pytorch
       run: |
-        python -m pip install torch==2.2.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
+        python -m pip install torch==2.3.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
     - name: Install package
       run: |
         python -m pip install --no-build-isolation .