From 0a3b364bbcee7269fd3d47793e1286712860dff6 Mon Sep 17 00:00:00 2001 From: Ze Mao Date: Tue, 3 Dec 2024 15:33:20 -0800 Subject: [PATCH] feat: Introduce cache_key for cache key customization Signed-off-by: Ze Mao --- api/v2alpha1/pipeline_spec.proto | 7 +++++-- api/v2alpha1/python/setup.py | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/api/v2alpha1/pipeline_spec.proto b/api/v2alpha1/pipeline_spec.proto index 075913f8052..5393b2e222b 100644 --- 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 index dccd702d231..12f3cacd588 100644 --- 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,