pod template inplace operations (#2899)

Signed-off-by: Daniel Sola <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>

flytekit/core/utils.py

@@ -1,3 +1,4 @@
+import copy
 import datetime
 import inspect
 import os
@@ -147,6 +148,9 @@ def _serialize_pod_spec(
 
     if pod_template.pod_spec is None:
         return {}
+
+    pod_template = copy.deepcopy(pod_template)
+
     containers = cast(V1PodSpec, pod_template.pod_spec).containers
     primary_exists = False
 

tests/flytekit/unit/core/test_pod_template_mutation.py

@@ -0,0 +1,28 @@
+from flytekit import PodTemplate, task
+from flytekit.configuration import SerializationSettings, ImageConfig, Image
+
+
+def test_mutation():
+    pod_template_before = PodTemplate(
+        labels={"lKeyA": "lValA", "lKeyB": "lValB"},
+    )
+
+    pod_template_after = PodTemplate(
+        labels={"lKeyA": "lValA", "lKeyB": "lValB"},
+    )
+
+    settings = SerializationSettings(image_config=ImageConfig(default_image=Image(
+                name='default',
+                fqn='cr.flyte.org/flyteorg/flytekit',
+                tag='latest',
+                digest=None
+            )))
+
+    @task(pod_template=pod_template_before)
+    def my_task():
+        pass
+
+    my_task.get_k8s_pod(settings=settings)
+
+
+    assert pod_template_before == pod_template_after