ML repa 2021 Flyte demo 2

repa_demo2.py

from flytekit import task, workflow


@task
def combine_strings(a: str, b: str) -> str:
    return f"{a} {b}"


@workflow
def hello_world(wf_in1: str, wf_in2: str) -> str:
    return combine_strings(a=wf_in1, b=wf_in2)


if __name__ == "__main__":
    print(hello_world(wf_in1="hello", wf_in2="world"))