Arguments mapping is very verbose depending on use case

[flaviuvadan]

Arguments mapping is very verbose depending on use case

[elliotgunton]

@flaviuvadan / @samj1912 this looks like a good improvement we should do for the new decorators - currently this isn't possible:

class PassthroughIO(Input, Output):
    my_str: str
    my_int: int
    my_artifact: Annotated[str, Artifact(name="my-artifact", loader=ArtifactLoader.json)]


@w.script()
def give_output() -> PassthroughIO:
    return PassthroughIO(my_str="test", my_int=42)


@w.script()
def take_input(inputs: PassthroughIO) -> Output:
    return Output(result=f"Got a string: {inputs.my_str}, got an int: {inputs.my_int}")


@w.dag()
def my_dag():
    output_task = give_output()
    take_input(output_task)

As you have to do

    take_input(
        PassthroughIO(
            my_str=output_task.my_str,
            my_int=output_task.my_int,
            my_artifact=output_task.my_artifact,
        )
    )

[samj1912]

Makes sense to me