fix slurm integration test (#949)

* Update dev-requirements.txt, Add docker-compose This fixed the slurm integration test. * Update slurm-local-integration-tests.yaml * Update dev-requirements.txt * Update slurm-local-integration-tests.yaml * Update dev-requirements.txt * Update slurm-local-integration-tests.yaml * Update slurmdockerint.sh * Update slurmdockerint.sh * Update slurm-local-integration-tests.yaml * Update slurmdockerint.sh * Update slurm-local-integration-tests.yaml
pytorch · Sep 10, 2024 · ce17fbb · ce17fbb
1 parent e1eb939
commit ce17fbb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/slurm-local-integration-tests.yaml b/.github/workflows/slurm-local-integration-tests.yaml
@@ -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/[email protected]
@@ -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: |

diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -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