#592 Check for None values in branch nodes
Pull request opened by
pingsutw
TL;DR
blocked by
flyteorg/flytekit#1747
Add support comparison (EQ and NEQ) between nil and non-nil values in the branch nodes.
Type
☐ Bug Fix
☑︎ Feature
☐ Plugin
Are all requirements met?
☑︎ Code completed
☑︎ Smoke tested
☐ Unit tests added
☐ Code documentation added
☐ Any pending items have an associated Issue
Complete description
import random
import typing
from flytekit import conditional, task, workflow
@task
def coin_toss(seed: int) -> typing.Optional[bool]:
r = random.Random(seed)
if r.random() < 0.5:
return True
return None
@task
def failed() -> int:
return -1
@task
def success() -> int:
return 0
@workflow
def wf(seed: int = 5) -> int:
result = coin_toss(seed=seed)
return conditional("test").if_(result.is_none()).then(success()).else_().then(failed())
if __name__ == '__main__':
wf()
Tracking Issue
flyteorg/flyte#3514
Follow-up issue
NA
flyteorg/flytepropeller
GitHub Actions: Docker Build Images / Build Docker Image
GitHub Actions: Check Go Generate / Go Generate
GitHub Actions: Unit Tests / Run Unit Test
GitHub Actions: Lint / Run Lint
GitHub Actions: Analyze (go)
✅ 2 other checks have passed
2/7 successful checks