make 'setup-storage-initializer-and-trainer' executable

Signed-off-by: helenxie-bit <[email protected]>
kubeflow · Sep 21, 2024 · f217794 · f217794
1 parent 45eb7e0
commit f217794
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/.github/workflows/e2e-test-train-api.yaml b/.github/workflows/e2e-test-train-api.yaml
@@ -48,7 +48,7 @@ jobs:
           GANG_SCHEDULER_NAME: "none"
           KUBERNETES_VERSION: ${{ matrix.kubernetes-version }}
 
-      - name: Build and load storage initializer and trainer
+      - name: Build storage initializer and trainer
         run: |
           ./scripts/gha/setup-storage-initializer-and-trainer.sh
         env:

diff --git a/scripts/gha/setup-storage-initializer-and-trainer.sh b/scripts/gha/setup-storage-initializer-and-trainer.sh
diff --git a/sdk/python/kubeflow/training/api/training_client.py b/sdk/python/kubeflow/training/api/training_client.py
@@ -258,9 +258,9 @@ def train(
             ],
             volume_mounts=[constants.STORAGE_INITIALIZER_VOLUME_MOUNT],
         )
-        base_image1=os.getenv(
-                "STORAGE_INITIALIZER_IMAGE", constants.STORAGE_INITIALIZER_IMAGE_DEFAULT
-            )
+        base_image1 = os.getenv(
+            "STORAGE_INITIALIZER_IMAGE", constants.STORAGE_INITIALIZER_IMAGE_DEFAULT
+        )
         print("base_image1: " + base_image1)
 
         # create app container spec
@@ -291,10 +291,10 @@ def train(
             volume_mounts=[constants.STORAGE_INITIALIZER_VOLUME_MOUNT],
             resources=resources_per_worker,
         )
-        base_image2=os.getenv(
-                "TRAINER_TRANSFORMER_IMAGE_DEFAULT",
-                constants.TRAINER_TRANSFORMER_IMAGE_DEFAULT,
-            )
+        base_image2 = os.getenv(
+            "TRAINER_TRANSFORMER_IMAGE_DEFAULT",
+            constants.TRAINER_TRANSFORMER_IMAGE_DEFAULT,
+        )
         print("base_image2: " + base_image2)
 
         storage_initializer_volume = models.V1Volume(