[BUG] Task chaining does not work inside local execution of a workflow

[ggydush-fn]

Describe the bug

Running a workflow locally gives a TypeError when using task-chaining syntax.

Expected behavior

Workflow runs locally without error.

Additional context to reproduce

from flytekit import task, workflow


@task
def task_a(x: int) -> int:
    return x


@task
def task_b(x: int) -> int:
    return x


@workflow
def test(x: int) -> int:
    a = task_a(x=x)
    b = task_b(x=x)
    a >> b
    return a


if __name__ == "__main__":
    test(x=1)

Raises:

TypeError: unsupported operand type(s) for >>: 'Promise' and 'Promise'

Screenshots

No response

Are you sure this issue hasn't been raised already?

• Yes

Have you read the Code of Conduct?

• Yes