kubeflow_pipelines_hello_world_component_and_compiler.py

1import kfp
2from kfp import dsl
3
4# Define a simple component that takes a string and prints a greeting
5@dsl.component
6def say_hello(name: str) -> str:
7    greeting = f'Hello, {name}!'
8    print(greeting)
9    return greeting
10
11# Define a pipeline that uses the component
12@dsl.pipeline(
13    name='hello-world-pipeline',
14    description='A simple pipeline that says hello.'
15)
16def hello_pipeline(recipient: str = 'World'):
17    hello_task = say_hello(name=recipient)
18
19if __name__ == '__main__':
20    # Compile the pipeline to a YAML file
21    kfp.compiler.Compiler().compile(
22        pipeline_func=hello_pipeline,
23        package_path='hello_world_pipeline.yaml'
24    )