add ContainerTask cache example code (#1758)

Signed-off-by: Alex Wu <[email protected]>
flyteorg · Oct 23, 2024 · 0b350cd · 0b350cd
1 parent 135ae99
commit 0b350cd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/examples/customizing_dependencies/customizing_dependencies/raw_container.py b/examples/customizing_dependencies/customizing_dependencies/raw_container.py
@@ -1,6 +1,7 @@
 import logging
 
 from flytekit import ContainerTask, kwtypes, task, workflow
+from flytekit.core.base_task import TaskMetadata
 
 logger = logging.getLogger(__file__)
 
@@ -25,6 +26,7 @@
         "{{.inputs.b}}",
         "/var/outputs",
     ],
+    metadata=TaskMetadata(cache=True, cache_version="1.0"),
 )
 
 calculate_ellipse_area_python = ContainerTask(
@@ -41,6 +43,7 @@
         "{{.inputs.b}}",
         "/var/outputs",
     ],
+    metadata=TaskMetadata(cache=True, cache_version="1.0"),
 )
 
 calculate_ellipse_area_r = ContainerTask(
@@ -58,6 +61,7 @@
         "{{.inputs.b}}",
         "/var/outputs",
     ],
+    metadata=TaskMetadata(cache=True, cache_version="1.0"),
 )
 
 calculate_ellipse_area_haskell = ContainerTask(
@@ -73,6 +77,7 @@
         "{{.inputs.b}}",
         "/var/outputs",
     ],
+    metadata=TaskMetadata(cache=True, cache_version="1.0"),
 )
 
 calculate_ellipse_area_julia = ContainerTask(
@@ -89,6 +94,7 @@
         "{{.inputs.b}}",
         "/var/outputs",
     ],
+    metadata=TaskMetadata(cache=True, cache_version="1.0"),
 )