fix: update with_copy with _update_attribute

Signed-off-by: mao3267 <[email protected]>
flyteorg · Sep 4, 2024 · 97d4192 · 97d4192
1 parent 9a9707d
commit 97d4192
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/flytekit/image_spec/image_spec.py b/flytekit/image_spec/image_spec.py
@@ -272,12 +272,7 @@ def with_copy(self, src: List[str]) -> "ImageSpec":
         """
         Builder that returns a new image spec with the source files copied to the destination directory.
         """
-        new_image_spec = copy.deepcopy(self)
-        if new_image_spec.copy is None:
-            new_image_spec.copy = []
-
-        new_image_spec.copy.append(src)
-
+        new_image_spec = self._update_attribute("copy", src)
         return new_image_spec
 
     def force_push(self) -> "ImageSpec":