From 083e15572dc7982de18519505399bc89f27fcec3 Mon Sep 17 00:00:00 2001 From: helenxie-bit Date: Fri, 20 Sep 2024 22:40:03 -0700 Subject: [PATCH] separate step of loading images Signed-off-by: helenxie-bit --- .github/workflows/e2e-test-train-api.yaml | 22 +++++++++++++++++++--- 1 file changed, 19 insertions(+), 3 deletions(-) diff --git a/.github/workflows/e2e-test-train-api.yaml b/.github/workflows/e2e-test-train-api.yaml index 1df822a735..198fdd7f1c 100644 --- 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