Brandon Segal
07/21/2023, 5:10 PMKevin Su
07/21/2023, 5:29 PMctx.user_space_params._attrs["GET_ORIGINAL_TASK"] = True
Brandon Segal
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
Kevin Su
07/21/2023, 5:40 PMBrandon Segal
07/21/2023, 5:41 PMKevin Su
07/21/2023, 5:50 PMpyflyte run --env
Brandon Segal
07/21/2023, 5:55 PMKetan (kumare3)