Mike Ossareh
05/23/2022, 4:46 PMfrom our.codebase.task import task, OurParameterType, OurTaskResult
def test_our_task():
@workflow
def do_workflow(arg1: FlyteDirectory, arg2: OurParameterType) -> OurTaskResult:
return task(arg1, arg2)
arg1 = foo
arg2 = bar
ret: OurTaskResult = do_workflow(arg1, arg2)
no parameter named arg2
2. VoidPromise is incompatible with OurTaskResult
do_work
be something else?OurTaskResult
is essentially a:
@dataclass_json
@dataclass
class FlyteResult:
result: <our dc>
Where our dc
is passed during module evaluationEduardo Apolinario (eapolinario)
05/23/2022, 5:07 PMdo_workflow
with:
do_workflow(arg1=arg1, arg2=arg2)
As for #2, this is a byproduct of the flytekit decorators. I think we can do something there, but will require a few changes.Mike Ossareh
05/23/2022, 5:18 PMEduardo Apolinario (eapolinario)
05/23/2022, 5:21 PMMike Ossareh
05/23/2022, 5:21 PMEduardo Apolinario (eapolinario)
05/23/2022, 5:22 PMMike Ossareh
05/23/2022, 5:22 PMEduardo Apolinario (eapolinario)
05/23/2022, 5:22 PMMike Ossareh
05/23/2022, 5:25 PMEduardo Apolinario (eapolinario)
05/23/2022, 5:25 PMMike Ossareh
05/23/2022, 5:56 PMEduardo Apolinario (eapolinario)
05/23/2022, 7:29 PM