Brian O'Donovan
05/09/2023, 4:12 PMmy_shelltask = ShellTask(
name="the_shelltask",
debug = True,
script="""
runs some shell code with {inputs.input1} and {inputs.input2}
""",
inputs=kwtypes(
input1=some_value,
input2=some_other_value
),
output_locs=[
OutputLocation(var='output1', var_type=FlyteFile, location="/tmp/output1.file"),
OutputLocation(var='output2', var_type=FlyteFile, location="/tmp/output2.file")
]
)
@task
def run_shelltask(MyDataClass, input1 = input1, input2 = input2) -> Tuple[MyDataClass, FlyteFile, FlyteFile]:
output1, output2 = my_shelltask(input1=input1)
return MyDataClass, output1, output2
Ketan (kumare3)
Brian O'Donovan
05/10/2023, 3:34 AMKetan (kumare3)