Introduces PipelineConfig class with TTL configuration

Signed-off-by: Ricardo M. Oliveira <[email protected]>
kubeflow · Oct 31, 2024 · 961daf4 · 961daf4
1 parent 6684b6d
commit 961daf4
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 11 deletions.
diff --git a/api/v2alpha1/go/cachekey/cache_key.pb.go b/api/v2alpha1/go/cachekey/cache_key.pb.go
diff --git a/api/v2alpha1/go/pipelinespec/pipeline_spec.pb.go b/api/v2alpha1/go/pipelinespec/pipeline_spec.pb.go
diff --git a/api/v2alpha1/pipeline_spec.proto b/api/v2alpha1/pipeline_spec.proto
@@ -1104,4 +1104,5 @@ message PlatformDeploymentConfig {
 // Spec for pipeline-level config options. See PipelineConfig DSL class.
 message PipelineConfig {
   // TODO add pipeline-level configs
+  int32 pipelineTtl = 1;
 }
diff --git a/sdk/python/kfp/compiler/pipeline_spec_builder.py b/sdk/python/kfp/compiler/pipeline_spec_builder.py
@@ -2077,6 +2077,11 @@ def _merge_pipeline_config(pipelineConfig: pipeline_config.PipelineConfig,
     #         '<some pipeline config option>': pipelineConfig.<get that value>,
     #     }}, platformSpec.platforms['kubernetes'])
 
+    pipeline_config_json = json_format.ParseDict(
+        {'pipelineConfig': {
+            'pipelineTtl': pipelineConfig.get_ttl(),
+        }}, platformSpec.platforms['kubernetes'])
+
     return platformSpec
 
 

diff --git a/sdk/python/kfp/dsl/pipeline_config.py b/sdk/python/kfp/dsl/pipeline_config.py
@@ -17,7 +17,9 @@
 class PipelineConfig:
     """PipelineConfig contains pipeline-level config options."""
 
-    def __init__(self):
-        pass
+    def __init__(self, ttl=None):
+        self.ttl, ttl
 
     # TODO add pipeline level configs
+    def set_pipeline_ttl(self, ttl: int):
+        self.ttl = ttl