Update pr-gpu.yaml (mosaicml#1164)

ShashankMosaicML · May 2, 2024 · 0d58f46 · 0d58f46
1 parent 124de4c
commit 0d58f46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -28,7 +28,7 @@ jobs:
           container: mosaicml/pytorch:2.3.0_cu121-python3.11-ubuntu20.04
           markers: "gpu"
           pytest_command: "coverage run -m pytest"
-          deps_group: "all"
+          pip_deps: "[all]"
     name: ${{ matrix.name }}
     if: github.repository_owner == 'mosaicml'
     with: