Minor fix.

Signed-off-by: Revital Sur <[email protected]>
IBM · Feb 10, 2025 · 8468356 · 8468356
1 parent ce8e919
commit 8468356
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/transforms/code/proglang_select/kfp_ray/proglang_select_wf.py b/transforms/code/proglang_select/kfp_ray/proglang_select_wf.py
@@ -217,7 +217,10 @@ def lang_select(
             # As a workaround, the secret name is hard coded.
             env2key = ComponentUtils.set_secret_key_to_env(prefix=PREFIX)
             kubernetes.use_secret_as_env(task=execute_job, secret_name=S3_SECRET, secret_key_to_env=env2key)
+            env2key = ComponentUtils.set_secret_key_to_env()
+            kubernetes.use_secret_as_env(task=execute_job, secret_name=S3_SECRET, secret_key_to_env=env2key)
         else:
+            ComponentUtils.set_s3_env_vars_to_component(execute_job, data_s3_access_secret)
             ComponentUtils.set_s3_env_vars_to_component(execute_job, proglang_select_s3_access_secret, prefix=PREFIX)
         execute_job.after(ray_cluster)