From 198d3eb0b559c359837eab3137ceac6ad3f09b74 Mon Sep 17 00:00:00 2001
From: Brian <23239305+b-chu@users.noreply.github.com>
Date: Sun, 1 Sep 2024 01:41:10 -0400
Subject: [PATCH] Add container

---
 .github/workflows/coverage.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/coverage.yaml b/.github/workflows/coverage.yaml
index 5ca86861fb..75643e7a57 100644
--- a/.github/workflows/coverage.yaml
+++ b/.github/workflows/coverage.yaml
@@ -9,6 +9,7 @@ jobs:
   coverage:
     timeout-minutes: 5
     runs-on: ubuntu-latest
+    container: mosaicml/pytorch:2.4.0_cpu-python3.11-ubuntu20.04
     steps:
     - name: Checkout Repo
       uses: actions/checkout@v3