可在workflow內,設定條件,並呼叫對應task進行計算。
import flytekit as fl
import random
@fl.task
def get_num() -> float:
return random.uniform(0.0, 1000.0)
@fl.task
def win_msg() -> str:
return "you win"
@fl.task
def lose_msg() -> str:
return "you lose"
@fl.workflow
def num_wf(my_num: float) -> str:
result = get_num()
return fl.conditional("compare").if_(result > my_num).then(win_msg()).else_().then(lose_msg())