feat: Introduce cache_key for cache key customization (#11434)

Signed-off-by: Ze Mao <[email protected]> Co-authored-by: Ze Mao <[email protected]>
kubeflow · Dec 5, 2024 · 50b367f · 50b367f
1 parent b4799df
commit 50b367f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/api/v2alpha1/pipeline_spec.proto b/api/v2alpha1/pipeline_spec.proto
@@ -461,6 +461,9 @@ message PipelineTaskSpec {
   message CachingOptions {
     // Whether or not to enable cache for this task. Defaults to false.
     bool enable_cache = 1;
+    // Customized cache key for this task. If set, the cache_key will be used
+    // as the key for the task's cache.
+    string cache_key = 2;
   }
   CachingOptions caching_options = 6;
 
@@ -620,7 +623,7 @@ message PipelineInfo {
   // Required field. The name of the pipeline.
   // The name will be used to create or find pipeline context in MLMD.
   string name = 1;
-  
+
   // Optional fields. The readable display name for the pipeline template.
   // Should not exceed 1024 characters.
   string display_name = 2;
@@ -783,7 +786,7 @@ message PipelineDeploymentConfig {
         string resource_count = 4;
       }
       AcceleratorConfig accelerator = 3;
-      
+
       reserved 4;
     }
     ResourceSpec resources = 5;

diff --git a/api/v2alpha1/python/setup.py b/api/v2alpha1/python/setup.py
@@ -15,7 +15,7 @@
 import setuptools
 
 NAME = 'kfp-pipeline-spec'
-VERSION = '0.5.0'
+VERSION = '0.6.0'
 
 setuptools.setup(
     name=NAME,