From c69c791966b019f65d54fb0aa85ee4141029f81d Mon Sep 17 00:00:00 2001
From: rkpattnaik780 <rkpattnaik780@gmail.com>
Date: Tue, 9 Apr 2024 22:54:29 +0530
Subject: [PATCH] fix: remove helper method to set image pull policy

---
 .../kubeflow/v2/python_dsl_template.jinja2        | 15 ---------------
 1 file changed, 15 deletions(-)

diff --git a/elyra/templates/kubeflow/v2/python_dsl_template.jinja2 b/elyra/templates/kubeflow/v2/python_dsl_template.jinja2
index e7f80fe7c..45272bf89 100644
--- a/elyra/templates/kubeflow/v2/python_dsl_template.jinja2
+++ b/elyra/templates/kubeflow/v2/python_dsl_template.jinja2
@@ -72,18 +72,6 @@ def add_pod_annotation(
 
     return task
 
-def set_image_pull_policy(task: PipelineTask, policy: str) -> PipelineTask:
-
-    if policy not in ['Always', 'Never', 'IfNotPresent']:
-        raise ValueError(
-            'Invalid imagePullPolicy. Must be one of `Always`, `Never`, `IfNotPresent`.'
-        )
-    msg = common.get_existing_kubernetes_config_as_message(task)
-    msg.image_pull_policy = policy
-    task.platform_config['kubernetes'] = json_format.MessageToDict(msg)
-
-    return task
-
 # ------------------------------------------------------------------
 # end of missing functions
 # ------------------------------------------------------------------
@@ -128,9 +116,6 @@ def generated_pipeline(
 {%     endif %}
 {%  endfor %}
     )
-{%  if workflow_task.task_modifiers.image_pull_policy %}
-    set_image_pull_policy({{ task_name }}, "{{ workflow_task.task_modifiers.image_pull_policy }}")
-{%  endif %}
 {%  if workflow_task.task_modifiers.object_storage_secret %}
     secret.use_secret_as_env({{ task_name }}, "{{ workflow_task.task_modifiers.object_storage_secret }}", { "AWS_ACCESS_KEY_ID": "AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY": "AWS_SECRET_ACCESS_KEY" })
 {%  endif %}