Set spec.description when compiling to Kubernetes manifests

Signed-off-by: mprahl <[email protected]>

Author: mprahl

sdk/python/kfp/compiler/compiler_test.py

@@ -1039,7 +1039,8 @@ def simple_pipeline():
     def test_compile_with_kubernetes_manifest_format(self):
         with tempfile.TemporaryDirectory() as tmpdir:
 
-            @dsl.pipeline(name='my-pipeline')
+            @dsl.pipeline(
+                name='my-pipeline', description='A simple test pipeline')
             def my_pipeline(input1: str):
                 print_op(message=input1)
 
@@ -1080,6 +1081,8 @@ def my_pipeline(input1: str):
                              pipeline_name)
             self.assertEqual(pipeline_manifest['spec']['displayName'],
                              pipeline_display_name)
+            self.assertEqual(pipeline_manifest['spec']['description'],
+                             'A simple test pipeline')
             self.assertEqual(pipeline_manifest['metadata']['namespace'],
                              namespace)
 
@@ -1091,6 +1094,8 @@ def my_pipeline(input1: str):
                              pipeline_version_name)
             self.assertEqual(pipeline_version_manifest['spec']['displayName'],
                              pipeline_version_display_name)
+            self.assertEqual(pipeline_version_manifest['spec']['description'],
+                             'A simple test pipeline')
             self.assertEqual(pipeline_version_manifest['spec']['pipelineName'],
                              pipeline_name)
             self.assertEqual(pipeline_version_manifest['metadata']['namespace'],

sdk/python/kfp/compiler/pipeline_spec_builder.py

@@ -2035,6 +2035,7 @@ def write_pipeline_spec_to_file(
             opts=opts,
             pipeline_spec=pipeline_spec,
             platform_spec=platform_spec,
+            pipeline_description=pipeline_description,
         )
         return
 
@@ -2077,6 +2078,7 @@ def _write_kubernetes_manifest_to_file(
     opts: KubernetesManifestOptions,
     pipeline_spec: pipeline_spec_pb2.PipelineSpec,
     platform_spec: pipeline_spec_pb2.PlatformSpec,
+    pipeline_description: Union[str, None] = None,
 ) -> None:
     pipeline_name = opts.pipeline_name
     pipeline_display_name = opts.pipeline_display_name
@@ -2103,6 +2105,8 @@ def _write_kubernetes_manifest_to_file(
                 'displayName': pipeline_display_name,
             },
         }
+        if pipeline_description:
+            pipeline_manifest['spec']['description'] = pipeline_description
         documents.append(pipeline_manifest)
 
     # PipelineVersion manifest
@@ -2120,6 +2124,8 @@ def _write_kubernetes_manifest_to_file(
             'platformSpec': platform_spec_dict,
         },
     }
+    if pipeline_description:
+        pipeline_version_manifest['spec']['description'] = pipeline_description
     documents.append(pipeline_version_manifest)
 
     with open(package_path, 'w') as yaml_file: