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

Signed-off-by: Antonin Stefanutti <[email protected]>
kubeflow · Jan 7, 2025 · 855a879 · 855a879
1 parent cf74126
commit 855a879
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk_v2/kubeflow/training/utils/utils.py b/sdk_v2/kubeflow/training/utils/utils.py
@@ -166,8 +166,8 @@ def get_args_using_train_func(
                 read -r -d '' SCRIPT << EOM\n
                 {func_code}
                 EOM
-                printf "%s" \"$SCRIPT\" > \"{func_file}\"
-                {entrypoint} \"{func_file}\""""
+                printf "%s" \"$SCRIPT\" > \"$program_path\"/\"{func_file}\"
+                {entrypoint} \"$program_path\"/\"{func_file}\""""
     )
 
     # Add function code to the execute script.