Fix read permission denied on train script when run as non-root

Signed-off-by: Antonin Stefanutti <[email protected]>
kubeflow · Jan 9, 2025 · 5cabee9 · 5cabee9
1 parent cf74126
commit 5cabee9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/manifests/v2/base/runtimes/pre-training/torch-distributed.yaml b/manifests/v2/base/runtimes/pre-training/torch-distributed.yaml
@@ -31,3 +31,9 @@ spec:
                           env | grep PET_
 
                           pip list
+                      volumeMounts:
+                        - name: workspace
+                          mountPath: /workspace
+                  volumes:
+                    - name: workspace
+                      emptyDir: {}
diff --git a/sdk_v2/kubeflow/training/utils/utils.py b/sdk_v2/kubeflow/training/utils/utils.py
@@ -162,7 +162,6 @@ def get_args_using_train_func(
     # That allows to execute the training script with the entrypoint.
     exec_script = textwrap.dedent(
         """
-                program_path=$(mktemp -d)
                 read -r -d '' SCRIPT << EOM\n
                 {func_code}
                 EOM