separate step of loading images

Signed-off-by: helenxie-bit <[email protected]>
kubeflow · Sep 21, 2024 · 083e155 · 083e155
1 parent f217794
commit 083e155
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/.github/workflows/e2e-test-train-api.yaml b/.github/workflows/e2e-test-train-api.yaml
@@ -58,14 +58,30 @@ jobs:
       - name: Check disk space
         run: df -h
 
+      - name: Load storage initializer
+        run: |
+          kind load docker-image ${{ env.STORAGE_INITIALIZER_CI_IMAGE }} --name ${{ env.KIND_CLUSTER }}
+          docker rmi ${{ env.STORAGE_INITIALIZER_CI_IMAGE }}
+        env:
+          KIND_CLUSTER: training-operator-cluster
+          STORAGE_INITIALIZER_CI_IMAGE: kubeflowtraining/storage-initializer:test
+
+      - name: Check disk space
+        run: df -h
+
+      - name: Load trainer
+        run: |
+          kind load docker-image ${{ env.TRAINER_CI_IMAGE }} --name ${{ env.KIND_CLUSTER }}
+          docker rmi ${{ env.TRAINER_CI_IMAGE }}
+        env:
+          KIND_CLUSTER: training-operator-cluster
+          TRAINER_CI_IMAGE: kubeflowtraining/trainer:test
+
       - name: Run tests
         run: |
-          kind load docker-image ${{ env.STORAGE_INITIALIZER_IMAGE }} --name ${{ env.KIND_CLUSTER }}
-          kind load docker-image ${{ env.TRAINER_TRANSFORMER_IMAGE_DEFAULT }} --name ${{ env.KIND_CLUSTER }} 
           pip install pytest
           python3 -m pip install -e sdk/python[huggingface]
           pytest -s sdk/python/test/e2e-train-api/test_e2e_train_api.py --log-cli-level=debug
         env:
-          KIND_CLUSTER: training-operator-cluster
           STORAGE_INITIALIZER_IMAGE: kubeflowtraining/storage-initializer:test
           TRAINER_TRANSFORMER_IMAGE_DEFAULT: kubeflowtraining/trainer:test