diff --git a/.github/workflows/ci-sharktank.yml b/.github/workflows/ci-sharktank.yml index 859a4b507..f162579a1 100644 --- a/.github/workflows/ci-sharktank.yml +++ b/.github/workflows/ci-sharktank.yml @@ -28,9 +28,12 @@ jobs: matrix: python-version: ["3.11", "3.12"] torch-version: ["2.3.0", "2.4.1", "2.5.1"] - os: [ubuntu-24.04] + os: [Ubuntu] include: - - os: windows-2022 + - os: Ubuntu + runs-on: ubuntu-24.04 + - os: Windows + runs-on: windows-2022 python-version: "3.11" torch-version: "2.3.0" exclude: @@ -38,7 +41,7 @@ jobs: # `torch.compile` requires torch>=2.4.0 for Python 3.12+ torch-version: "2.3.0" fail-fast: false - runs-on: ${{matrix.os}} + runs-on: ${{matrix.runs-os}} defaults: run: shell: bash