Create a Steps template with pydantic models and hera runner

[bieniekmateusz]

The new hera runner together with pydantic models looks like this

@script()
def work(world: PydanticWorld) -> world: PydanticWorld:
   ....

But I don't know how to adapt this to create a Scripts template:

def create_template() -> Steps:
    with Steps( ..., inputs={world: PydanticWorld}, outputs={world: PydanticWorld})
        script 1 to 10

[bieniekmateusz]

Maybe even a nicer way to do this would be if there was a decorator that harvested the inputs, outputs in the same way it does for @script:

@Steps
def template(world: PydanticWorld) -> world: PydanticWorld:
    r1 = script1(arguments={"world": world})
    r2 = script2(arguments={"world": r1.result})
    return r2.result   

[bieniekmateusz]

I found a workaround by employing Parameters that I assume pass the serialised objects. Here is the example:

def template(name="pydantic-scripts") -> Steps:

    with Steps(
        name=name,
        inputs=[hwf.Parameter(name='pmodels')]
    ) as steps:

        work1_done = work1(
            name=...
            arguments={
                "pmodel": "{{item}}",
            },
           # pmodels here are split and automatically converted into the appropriate pydantic model
            with_param=steps.get_parameter('pmodels').value,
        )

        work2_res = work2(
            name=...
            arguments={"pmodels": work1_done.result}
        )

    return steps

This is clearly a workaround but I'll leave this open until a more "consistent" approach is shared.

[sambhav]

@bieniekmateusz your suggestion already exists - see https://hera.readthedocs.io/en/stable/user-guides/decorators/#using-decorators