diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index 2c4aeb3d4c..16589cf4c6 100644 --- 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: