fix container

mosaicml · Aug 31, 2024 · 8fa3476 · 8fa3476
1 parent c60c0cf
commit 8fa3476
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.github/workflows/pr-cpu.yaml b/.github/workflows/pr-cpu.yaml
@@ -17,6 +17,7 @@ jobs:
   pytest-cpu:
     name: ${{ matrix.name }}
     runs-on: ubuntu-latest
+    container: ${{ matrix.container }}
     strategy:
       matrix:
         include:

diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -18,6 +18,7 @@ jobs:
     name: ${{ matrix.name }}
     if: github.repository_owner == 'mosaicml'
     runs-on: linux-ubuntu-latest
+    container: ${{ matrix.container }}
     strategy:
       fail-fast: false
       matrix: