From b9b17aae51182824f1c6960634bdf42fa065560b Mon Sep 17 00:00:00 2001
From: Mihir Patel <mihir.v.patel7@gmail.com>
Date: Fri, 30 Aug 2024 17:32:02 -0400
Subject: [PATCH] rerun tests

---
 .github/workflows/pr-cpu.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/pr-cpu.yaml b/.github/workflows/pr-cpu.yaml
index f274d8a1be..a982669cdd 100644
--- 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:
@@ -30,7 +31,6 @@ jobs:
       uses: mosaicml/ci-testing/.github/actions/pytest-cpu@debug-2
       with:
         name: ${{ matrix.name }}
-        container: ${{ matrix.container }}
         pip_deps: ${{ matrix.pip_deps }}
         pytest_command: ${{ matrix.pytest_command }}
         pytest_markers: ${{ matrix.markers }}