acoustic-carpenter-78188
05/10/2023, 6:26 PMfrom flytekit import kwtypes, task
from flytekit.extras.tasks.shell import OutputLocation, ShellTask
from flytekit.types.file import FlyteFile
st = ShellTask(
name="shell_task",
debug=True,
script="""
touch {inputs.filename}
""",
inputs=kwtypes(filename=str),
output_locs=[OutputLocation(var="outfile", var_type=FlyteFile, location="{inputs.filename}")],
)
@task
def t1_task( ) -> FlyteFile:
outfile = st(filename='test.txt')
return outfile
if __name__ == "__main__":
print(f"let's run a shelltask and return a FlyteFile")
outfile = t1_task()
flyteorg/flyte