bored-laptop-29637
07/21/2023, 5:10 PMglamorous-carpet-83516
07/21/2023, 5:29 PMctx.user_space_params._attrs["GET_ORIGINAL_TASK"] = True
bored-laptop-29637
07/21/2023, 5:37 PM@task
def task_1(execution_id:str):
ctx = FlyteContextManager.get_current_context()
ctx.user_space_params._attrs["GET_ORIGINAL_TASK"] = True
@task
def task_2():
ctx = FlyteContextManager.get_current_context()
print(ctx.user_space_params._attrs["GET_ORIGINAL_TASK"])
@workflow()
def wf(execution_id:str)->None:
task_1_output = task_1()
task_2_output = task_2()
task_1_output >> task_2_output
glamorous-carpet-83516
07/21/2023, 5:40 PMbored-laptop-29637
07/21/2023, 5:41 PMglamorous-carpet-83516
07/21/2023, 5:50 PMpyflyte run --env
glamorous-carpet-83516
07/21/2023, 5:51 PMbored-laptop-29637
07/21/2023, 5:55 PMfreezing-airport-6809