Update one more file

dmlc · Jan 11, 2025 · 7b42c18 · 7b42c18
1 parent cc839cc
commit 7b42c18
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -220,22 +220,22 @@ jobs:
       matrix:
         include:
           - description: single-gpu
-            container: xgb-ci.gpu
+            image_repo: xgb-ci.gpu
             suite: gpu
             runner: linux-amd64-gpu
             artifact_from: build-cuda
           - description: multiple-gpu
-            container: xgb-ci.gpu
+            image_repo: xgb-ci.gpu
             suite: mgpu
             runner: linux-amd64-mgpu
             artifact_from: build-cuda
           - description: cpu-amd64
-            container: xgb-ci.cpu
+            image_repo: xgb-ci.cpu
             suite: cpu
             runner: linux-amd64-cpu
             artifact_from: build-cuda
           - description: cpu-arm64
-            container: xgb-ci.aarch64
+            image_repo: xgb-ci.aarch64
             suite: cpu-arm64
             runner: linux-arm64-cpu
             artifact_from: build-cpu-arm64
@@ -257,4 +257,4 @@ jobs:
           mv -v wheelhouse/xgboost .
           chmod +x ./xgboost
       - name: Run Python tests, ${{ matrix.description }}
-        run: bash ops/pipeline/test-python-wheel.sh ${{ matrix.suite }} ${{ matrix.container }}
+        run: bash ops/pipeline/test-python-wheel.sh ${{ matrix.suite }} ${{ matrix.image_repo }}