take a look at the snippet in the PR https://github.com/flyteorg/flytekit/pull/850

let me know if this still doesn’t help. we’ll continue working on the workflow part of it next week.