Skip to content
Merged
7 changes: 3 additions & 4 deletions .github/workflows/slurm-local-integration-tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ jobs:
set -ex

cd slurm-docker-cluster
docker-compose pull --ignore-pull-failures
docker compose pull --ignore-pull-failures
continue-on-error: true
# cache layers to avoid rebuilding slurm
- uses: satackey/action-docker-layer-caching@v0.0.11
Expand All @@ -48,11 +48,10 @@ jobs:
run: |
set -ex
cd slurm-docker-cluster
IMAGE_TAG=$SLURM_VERSION docker-compose up --detach
IMAGE_TAG=$SLURM_VERSION docker compose up --detach
- name: Install Dependencies
run:
run: |
set -ex

pip install wheel
- name: Run Slurm Integration Tests
run: |
Expand Down
1 change: 1 addition & 0 deletions dev-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ torchtext==0.17.1
torchvision==0.17.1
ts==0.5.1
ray[default]
wheel

# lint (linter versions are managed by lintrunner)
lintrunner
Expand Down
Loading