Skip to content

Commit 50b367f

Browse files
zzmaoZe Mao
andauthored
feat: Introduce cache_key for cache key customization (#11434)
Signed-off-by: Ze Mao <zemao@google.com> Co-authored-by: Ze Mao <zemao@google.com>
1 parent b4799df commit 50b367f

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

api/v2alpha1/pipeline_spec.proto

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -461,6 +461,9 @@ message PipelineTaskSpec {
461461
message CachingOptions {
462462
// Whether or not to enable cache for this task. Defaults to false.
463463
bool enable_cache = 1;
464+
// Customized cache key for this task. If set, the cache_key will be used
465+
// as the key for the task's cache.
466+
string cache_key = 2;
464467
}
465468
CachingOptions caching_options = 6;
466469

@@ -620,7 +623,7 @@ message PipelineInfo {
620623
// Required field. The name of the pipeline.
621624
// The name will be used to create or find pipeline context in MLMD.
622625
string name = 1;
623-
626+
624627
// Optional fields. The readable display name for the pipeline template.
625628
// Should not exceed 1024 characters.
626629
string display_name = 2;
@@ -783,7 +786,7 @@ message PipelineDeploymentConfig {
783786
string resource_count = 4;
784787
}
785788
AcceleratorConfig accelerator = 3;
786-
789+
787790
reserved 4;
788791
}
789792
ResourceSpec resources = 5;

api/v2alpha1/python/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import setuptools
1616

1717
NAME = 'kfp-pipeline-spec'
18-
VERSION = '0.5.0'
18+
VERSION = '0.6.0'
1919

2020
setuptools.setup(
2121
name=NAME,

0 commit comments

Comments
 (0)